fix type

pfnet · Oct 31, 2023 · 5e5d5d3 · 5e5d5d3
1 parent 4ffcf4d
commit 5e5d5d3
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
@@ -519,4 +519,4 @@ def _make_snapshot(self, manager: ExtensionsManagerProtocol) -> None:
         if self._rank == self._saver_rank:
             super()._make_snapshot(manager)
         if self._size > 1:
-            torch.distributed.barrier()
+            torch.distributed.barrier()  # type: ignore