Skip to content

Commit 5e2f952

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 844f61c commit 5e2f952

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/tests_pytorch/strategies/test_fsdp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,8 @@ def _make_trainer(max_epochs: int) -> Trainer:
298298
ckpt_path = Path(trainer.strategy.broadcast(str(tmp_path / "checkpoint")))
299299
trainer.save_checkpoint(ckpt_path)
300300

301-
assert ckpt_path.is_dir() and _is_sharded_checkpoint(ckpt_path)
301+
assert ckpt_path.is_dir()
302+
assert _is_sharded_checkpoint(ckpt_path)
302303

303304
# snapshot the trained local shards on this rank for a read-back comparison
304305
trained_params = deepcopy(list(trainer.model.parameters()))

0 commit comments

Comments
 (0)