Skip to content

Commit 0c73b63

Browse files
committed
update test
1 parent 31f1607 commit 0c73b63

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,8 @@ def test_loops_restore(tmpdir):
172172
ckpt_path = str(tmpdir / "last.ckpt")
173173

174174
trainer = Trainer(**trainer_args)
175+
trainer.strategy.connect(model)
176+
175177
for fn in TrainerFn:
176178
if fn != TrainerFn.TUNING:
177179
trainer_fn = getattr(trainer, f"{fn}_loop")

0 commit comments

Comments
 (0)