Update cunumeric/array.py

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Nov 9, 2023 · 329bf5a · 329bf5a
1 parent 2a2ead5
commit 329bf5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cunumeric/array.py b/cunumeric/array.py
@@ -3160,7 +3160,7 @@ def mean(
             if where_array is not None:
                 divisor = where_array._count_nonzero(axis=axis)
             else:
-                divisor = self.shape[axis]
+                divisor = np.array(self.shape[axis])
         # Divide by the number of things in the collapsed dimensions
         # Pick the right kinds of division based on the dtype
         sum_array = sum_array.astype(dtype)