Updated typehint for idist.all_gather method (#3089)

Description:
- This reflects the fact that idist can use all_gather_object internally

Author: vfdev-5

ignite/distributed/comp_models/base.py

@@ -214,7 +214,7 @@ def all_reduce(
 
     def all_gather(
         self, tensor: Union[torch.Tensor, float, str, Any], group: Optional[Any] = None
-    ) -> Union[torch.Tensor, float, List[float], List[str]]:
+    ) -> Union[torch.Tensor, float, List[float], List[str], List[Any]]:
         if not isinstance(tensor, (torch.Tensor, Number, str)):
             return self._do_all_gather_object(tensor, group=group)
 
@@ -355,11 +355,11 @@ def all_reduce(
         return tensor
 
     def all_gather(
-        self, tensor: Union[torch.Tensor, float, str], group: Optional[Any] = None
-    ) -> Union[torch.Tensor, float, List[float], List[str]]:
+        self, tensor: Union[torch.Tensor, float, str, Any], group: Optional[Any] = None
+    ) -> Union[torch.Tensor, float, List[float], List[str], List[Any]]:
         if isinstance(tensor, torch.Tensor):
             return tensor
-        return cast(Union[List[float], List[str]], [tensor])
+        return cast(Union[List[float], List[str], List[Any]], [tensor])
 
     def broadcast(
         self, tensor: Union[torch.Tensor, float, str, None], src: int = 0, safe_mode: bool = False

ignite/distributed/utils.py

@@ -351,8 +351,8 @@ def all_reduce(
 
 
 def all_gather(
-    tensor: Union[torch.Tensor, float, str], group: Optional[Union[Any, List[int]]] = None
-) -> Union[torch.Tensor, float, List[float], List[str]]:
+    tensor: Union[torch.Tensor, float, str, Any], group: Optional[Union[Any, List[int]]] = None
+) -> Union[torch.Tensor, float, List[float], List[str], List[Any]]:
     """Helper method to perform all gather operation.
 
     Args: