Update cunumeric/array.py

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Nov 9, 2023 · c52e0c5 · c52e0c5
1 parent a50f165
commit c52e0c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cunumeric/array.py b/cunumeric/array.py
@@ -3181,7 +3181,7 @@ def mean(
     @add_boilerplate()
     def _nanmean(
         self,
-        axis: Any = None,
+        axis: Optional[Union[int, tuple[int, ...]]] = None,
         dtype: Union[np.dtype[Any], None] = None,
         out: Union[ndarray, None] = None,
         keepdims: bool = False,