MAINT: avoid can_cast(pyscalar, dtype) NumPy 2 refuses it (for now)

rapidsai · May 31, 2024 · 9e5d611 · 9e5d611
1 parent b17af57
commit 9e5d611
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/cudf/cudf/core/_internals/where.py b/python/cudf/cudf/core/_internals/where.py
@@ -85,8 +85,10 @@ def _check_and_cast_columns_with_other(
             )
         return _normalize_categorical(source_col, other.astype(source_dtype))
 
-    if _is_non_decimal_numeric_dtype(source_dtype) and _can_cast(
-        other, source_dtype
+    if (
+        _is_non_decimal_numeric_dtype(source_dtype)
+        and not other_is_scalar  # can-cast fails for Python scalars
+        and _can_cast(other, source_dtype)
     ):
         common_dtype = source_dtype
     elif (