Skip to content

Commit 8ca5695

Browse files
committed
fix doctests
1 parent a300fe2 commit 8ca5695

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/lightning/pytorch/callbacks/early_stopping.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,8 @@ class EarlyStopping(Callback):
9090
Example::
9191
9292
>>> from lightning.pytorch import Trainer
93-
>>> from lightning.pytorch.callbacks import EarlyStopping, EarlyStoppingReason
93+
>>> from lightning.pytorch.callbacks import EarlyStopping
94+
>>> from lightning.pytorch.callbacks.early_stopping import EarlyStoppingReason
9495
>>> early_stopping = EarlyStopping('val_loss')
9596
>>> trainer = Trainer(callbacks=[early_stopping])
9697
>>> # After training...

0 commit comments

Comments
 (0)