Skip to content

Commit 058dd9f

Browse files
dtuitlexierule
authored andcommitted
Fix imports for lightning cli examples (#16871)
1 parent 5c3de46 commit 058dd9f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source-pytorch/cli/lightning_cli_intermediate_2.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ To support multiple models, when instantiating ``LightningCLI`` omit the ``model
5757
5858
# main.py
5959
from pytorch_lightning.cli import LightningCLI
60-
from pytorch_lightning.demos.boring_classes import DemoModel
60+
from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
6161
6262
6363
class Model1(DemoModel):
@@ -101,7 +101,7 @@ To support multiple data modules, when instantiating ``LightningCLI`` omit the `
101101
# main.py
102102
import torch
103103
from pytorch_lightning.cli import LightningCLI
104-
from pytorch_lightning.demos.boring_classes import BoringDataModule
104+
from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
105105
106106
107107
class FakeDataset1(BoringDataModule):

0 commit comments

Comments
 (0)