From 9d5b364698c239428297a309632069feae38668d Mon Sep 17 00:00:00 2001
From: Irina Demeshko <dirishka@gmail.com>
Date: Fri, 17 Nov 2023 13:21:04 -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 33233d284..411980d82 100644
--- a/cunumeric/array.py
+++ b/cunumeric/array.py
@@ -3141,7 +3141,7 @@ def mean(
 
         if axis is None:
             if where_array is not None:
-                divisor = np.array(where_array._count_nonzero())
+                divisor = where_array._count_nonzero()
             else:
                 divisor = np.array(reduce(lambda x, y: x * y, self.shape, 1))