From c52e0c58f436e6704265d6bddd473684b782ad4b Mon Sep 17 00:00:00 2001
From: Irina Demeshko <dirishka@gmail.com>
Date: Thu, 9 Nov 2023 11:26:08 -0800
Subject: [PATCH] Update cunumeric/array.py

Co-authored-by: Manolis Papadakis <manopapad@gmail.com>
---
 cunumeric/array.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cunumeric/array.py b/cunumeric/array.py
index 60e8cf2b2..0a362a166 100644
--- 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,