fix: only for numpy 2+

scikit-hep · Jun 20, 2024 · fa313c4 · fa313c4
1 parent c4c1be0
commit fa313c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/awkward/_reducers.py b/src/awkward/_reducers.py
@@ -62,7 +62,9 @@ def _dtype_for_kernel(cls, dtype: DTypeLike) -> DTypeLike:
         else:
             return dtype
 
-    _use32 = (ak._util.win or ak._util.bits32) and not ak._util.numpy2 and np.intp is np.int32
+    _use32 = ((ak._util.win or ak._util.bits32) and not ak._util.numpy2) or (
+        ak._util.numpy2 and np.intp is np.int32
+    )
 
     @classmethod
     def _promote_integer_rank(cls, given_dtype: DTypeLike) -> DTypeLike: