fix type

linshokaku · Dec 20, 2023 · 13e9f79 · 13e9f79
1 parent 6b07015
commit 13e9f79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_pfn_extras/training/extensions/_snapshot.py b/pytorch_pfn_extras/training/extensions/_snapshot.py
@@ -631,7 +631,7 @@ def _autoload(
         self, manager: ExtensionsManagerProtocol, loaded_fn: Optional[str]
     ) -> None:
         snapshot_dir = None if loaded_fn is None else os.path.dirname(loaded_fn)
-        snapshot_dir_list: list[Optional[str]] = [None] * self._size
+        snapshot_dir_list: List[Optional[str]] = [None] * self._size
         torch.distributed.all_gather_object(snapshot_dir_list, snapshot_dir)  # type: ignore[no-untyped-call]
         assert all(
             x == snapshot_dir_list[0] for x in snapshot_dir_list