Update cunumeric/array.py

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Nov 9, 2023 · c00661a · c00661a
1 parent 7070d9f
commit c00661a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cunumeric/array.py b/cunumeric/array.py
@@ -3154,7 +3154,7 @@ def mean(
             if where_array is not None:
                 divisor = where_array._count_nonzero()
             else:
-                divisor = reduce(lambda x, y: x * y, self.shape, 1)
+                divisor = np.array(reduce(lambda x, y: x * y, self.shape, 1))
 
         else:
             if where_array is not None: