From 0c73b635c2820a9ea26333c014c286e7de637dd5 Mon Sep 17 00:00:00 2001 From: rohitgr7 Date: Tue, 19 Jul 2022 23:51:44 +0530 Subject: [PATCH] update test --- .../trainer/connectors/test_checkpoint_connector.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py b/tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py index cf640f794f259..02e3221f4dfd4 100644 --- a/tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py +++ b/tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py @@ -172,6 +172,8 @@ def test_loops_restore(tmpdir): ckpt_path = str(tmpdir / "last.ckpt") trainer = Trainer(**trainer_args) + trainer.strategy.connect(model) + for fn in TrainerFn: if fn != TrainerFn.TUNING: trainer_fn = getattr(trainer, f"{fn}_loop")