From eb82e307c854631c8c17223acbe39968f7f477b5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Fri, 3 Feb 2023 01:52:16 +0100 Subject: [PATCH] Update import in bug report template (#16616) --- examples/pl_bug_report/bug_report_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pl_bug_report/bug_report_model.py b/examples/pl_bug_report/bug_report_model.py index 7739630237..551ea21721 100644 --- a/examples/pl_bug_report/bug_report_model.py +++ b/examples/pl_bug_report/bug_report_model.py @@ -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):