Made flake8 happy.

Lightning-AI · Mar 1, 2021 · 48ff9a3 · 48ff9a3
1 parent e400c35
commit 48ff9a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py
@@ -387,10 +387,9 @@ def format_checkpoint_name(self, epoch: int, step: int, metrics: Dict[str, Any],
             >>> ckpt = ModelCheckpoint(dirpath=tmpdir, filename='{epoch}-{val_loss:.2f}')
             >>> os.path.basename(ckpt.format_checkpoint_name(2, 3, metrics=dict(val_loss=0.123456)))
             'epoch=2-val_loss=0.12.ckpt'
-            >>> ckpt = ModelCheckpoint(
-            >>>     dirpath=tmpdir,
-            >>>     filename='epoch={epoch}-validation_loss={val_loss:.2f}',
-            >>>     auto_insert_metric_name=False)
+            >>> ckpt = ModelCheckpoint(dirpath=tmpdir,
+            ... filename='epoch={epoch}-validation_loss={val_loss:.2f}',
+            ... auto_insert_metric_name=False)
             >>> os.path.basename(ckpt.format_checkpoint_name(2, 3, metrics=dict(val_loss=0.123456)))
             'epoch=2-validation_loss=0.12.ckpt'
             >>> ckpt = ModelCheckpoint(dirpath=tmpdir, filename='{missing:d}')

diff --git a/tests/checkpointing/test_model_checkpoint.py b/tests/checkpointing/test_model_checkpoint.py
@@ -435,6 +435,7 @@ def test_model_checkpoint_format_checkpoint_name(tmpdir):
         auto_insert_metric_name=False)
     assert ckpt_name == 'epoch=003-val_acc=0.03'
 
+
 class ModelCheckpointExtensionTest(ModelCheckpoint):
     FILE_EXTENSION = '.tpkc'