diff --git a/cunumeric/array.py b/cunumeric/array.py index 411980d82..7a962d22b 100644 --- a/cunumeric/array.py +++ b/cunumeric/array.py @@ -3147,9 +3147,7 @@ def mean( else: if where_array is not None: - divisor = np.array( - where_array.sum(axis=axis, keepdims=keepdims) - ) + divisor = where_array.sum(axis=axis, keepdims=keepdims) else: divisor = np.array(self.shape[axis])