Fix pedantic type compatibility tests

manopapad · Jul 31, 2023 · 871c30b · 871c30b
1 parent f13afe9
commit 871c30b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cunumeric/_ufunc/ufunc.py b/cunumeric/_ufunc/ufunc.py
@@ -561,8 +561,9 @@ def _find_common_type(
         scalar_types = []
         array_types = []
         for arr, orig_arg in zip(arrs, orig_args):
-            if arr.ndim == 0 and not isinstance(orig_arg, ndarray):
-                scalar_types.append(orig_arg)
+            if arr.ndim == 0:
+                # Make sure all scalar arguments are NumPy arrays
+                scalar_types.append(np.asarray(orig_arg))
             else:
                 array_types.append(arr.dtype)