Update import in bug report template (#16616)
This commit is contained in:
parent
85f7e1c9c8
commit
eb82e307c8
|
@ -3,7 +3,7 @@ import os
|
|||
import torch
|
||||
from torch.utils.data import DataLoader, Dataset
|
||||
|
||||
from pytorch_lightning import LightningModule, Trainer
|
||||
from lightning.pytorch import LightningModule, Trainer
|
||||
|
||||
|
||||
class RandomDataset(Dataset):
|
||||
|
|
Loading…
Reference in New Issue