We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 844f61c commit 5e2f952Copy full SHA for 5e2f952
1 file changed
tests/tests_pytorch/strategies/test_fsdp.py
@@ -298,7 +298,8 @@ def _make_trainer(max_epochs: int) -> Trainer:
298
ckpt_path = Path(trainer.strategy.broadcast(str(tmp_path / "checkpoint")))
299
trainer.save_checkpoint(ckpt_path)
300
301
- assert ckpt_path.is_dir() and _is_sharded_checkpoint(ckpt_path)
+ assert ckpt_path.is_dir()
302
+ assert _is_sharded_checkpoint(ckpt_path)
303
304
# snapshot the trained local shards on this rank for a read-back comparison
305
trained_params = deepcopy(list(trainer.model.parameters()))
0 commit comments