From 13e9f79896ca614f5e2654a9e0d768b701a7701b Mon Sep 17 00:00:00 2001 From: Linsho Kaku Date: Wed, 20 Dec 2023 15:35:21 +0900 Subject: [PATCH] fix type --- pytorch_pfn_extras/training/extensions/_snapshot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_pfn_extras/training/extensions/_snapshot.py b/pytorch_pfn_extras/training/extensions/_snapshot.py index b4099271..263ea17a 100644 --- 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