typing

pydata · Sep 19, 2024 · 5540702 · 5540702
1 parent f64f967
commit 5540702
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/xarray/namedarray/_array_api/_manipulation_functions.py b/xarray/namedarray/_array_api/_manipulation_functions.py
@@ -356,7 +356,7 @@ def _set_dims(
         return permute_dims(out, dims=dim)
 
 
-def _broadcast_arrays(*arrays: NamedArray[Any, Any]):
+def _broadcast_arrays(*arrays: NamedArray[Any, Any]) -> NamedArray[Any, Any]:
     """
     TODO: Can this become xp.broadcast_arrays?
 
@@ -373,7 +373,9 @@ def _broadcast_arrays(*arrays: NamedArray[Any, Any]):
     return tuple(_set_dims(var, dims, shape) for var in arrays)
 
 
-def _broadcast_arrays_with_minimal_size(*arrays: NamedArray[Any, Any]):
+def _broadcast_arrays_with_minimal_size(
+    *arrays: NamedArray[Any, Any]
+) -> NamedArray[Any, Any]:
     """
     Given any number of variables, return variables with matching dimensions.
 
@@ -384,7 +386,9 @@ def _broadcast_arrays_with_minimal_size(*arrays: NamedArray[Any, Any]):
     return tuple(_set_dims(var, dims, None) for var in arrays)
 
 
-def _arithmetic_broadcast(x1: NamedArray[Any, Any], x2: NamedArray[Any, Any]):
+def _arithmetic_broadcast(
+    x1: NamedArray[Any, Any], x2: NamedArray[Any, Any]
+) -> NamedArray[Any, Any]:
     """
     Fu