Skip to content

Commit af23135

Browse files
committed
typing
1 parent 58e6028 commit af23135

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchft/manager.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -534,11 +534,11 @@ def _apply_pending_state_dict(self) -> None:
534534

535535
self._logger.info("applying pending state dict")
536536

537+
assert self._pending_state_dict is not None, "checkpoint was not staged"
537538
assert (
538539
self._load_state_dict is not None
539540
), "user load_state_dict is not initialized."
540541
self._load_state_dict(self._pending_state_dict["user"])
541-
assert self._pending_state_dict is not None, "checkpoint was not staged"
542542
self._pending_state_dict = None
543543
self._logger.info("Loaded state dict.")
544544

0 commit comments

Comments
 (0)