Fix typehint

pytorch · Feb 28, 2025 · 3240160 · 3240160
1 parent 7dbca9f
commit 3240160
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/distributed/utils.py b/ignite/distributed/utils.py
@@ -742,7 +742,7 @@ def download_dataset():
         barrier()
 
 
-def _rank_not_in_group(group: Optional[dist.ProcessGroup | "hvd.ProcessSet"]) -> bool:
+def _rank_not_in_group(group: Optional[Union[Any, List[int]]]) -> bool:
     """Check if the current process's rank is not in a given group."""
     if has_hvd_support and isinstance(group, hvd.common.process_sets.ProcessSet):
         return group.included()