Skip to content

Commit

Permalink
add docstring to test
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Sep 9, 2021
1 parent 8d97f7f commit 8f73fa8
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/loops/test_loops.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,11 @@ def test_connect_subloops(tmpdir):
assert new_batch_loop.trainer is trainer


def test_loop_restarting(tmpdir):
# TODO:
pass


class CustomException(Exception):
pass

Expand Down Expand Up @@ -649,6 +654,8 @@ def configure_optimizers_multiple(self):

@mock.patch.dict(os.environ, {"PL_FAULT_TOLERANT_TRAINING": "1"})
def test_fit_loop_reset(tmpdir):
"""Test that the reset logic in fit- and epoch loop is aware of whether the loop is restarting from a
completed loop or from a mid-epoch checkpoint."""

# generate checkpoints at end of epoch and mid-epoch
model = BoringModel()
Expand Down

0 comments on commit 8f73fa8

Please sign in to comment.