Skip to content

Commit 090fac8

Browse files
moi90Borda
authored andcommitted
Fix whitespace
1 parent f6a99d2 commit 090fac8

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

pytorch_lightning/callbacks/base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -87,4 +87,4 @@ def on_test_end(self, trainer, pl_module):
8787
pass
8888

8989
def on_interrupted(self, trainer, pl_module):
90-
"""Called when the training is interrupted by KeyboardInterrupt."""
90+
"""Called when the training is interrupted by KeyboardInterrupt."""

pytorch_lightning/trainer/callback_hook.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -104,4 +104,4 @@ def on_test_end(self):
104104
def on_interrupted(self):
105105
"""Called when the training is interrupted by KeyboardInterrupt."""
106106
for callback in self.callbacks:
107-
callback.on_interrupted(self, self.get_model())
107+
callback.on_interrupted(self, self.get_model())

tests/trainer/test_trainer.py

-2
Original file line numberDiff line numberDiff line change
@@ -653,7 +653,6 @@ def __init__(self):
653653
def on_interrupted(self, trainer, pl_module):
654654
self.exc_info = sys.exc_info()
655655

656-
657656
interrupt_callback = InterruptCallback()
658657
handle_interrupt_callback = HandleInterruptCallback()
659658

@@ -673,7 +672,6 @@ def on_interrupted(self, trainer, pl_module):
673672
assert isinstance(handle_interrupt_callback.exc_info[1], KeyboardInterrupt)
674673

675674

676-
677675
def test_gradient_clipping(tmpdir):
678676
"""
679677
Test gradient clipping

0 commit comments

Comments
 (0)