Handle NA cudf.Scalar in div by zero check

rapidsai · Nov 8, 2022 · f023aea · f023aea
1 parent 603fdbb
commit f023aea
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/python/cudf/cudf/core/column/numerical.py b/python/cudf/cudf/core/column/numerical.py
@@ -226,7 +226,14 @@ def _binaryop(self, other: ColumnBinaryOperand, op: str) -> ColumnBase:
                     and (tmp.dtype.type != np.bool_)
                     and (
                         (
-                            (np.isscalar(tmp) or isinstance(tmp, cudf.Scalar))
+                            (
+                                np.isscalar(tmp)
+                                or (
+                                    isinstance(tmp, cudf.Scalar)
+                                    # host to device copy
+                                    and (tmp.value is not pd.NA)
+                                )
+                            )
                             and (0 == tmp)
                         )
                         or ((isinstance(tmp, NumericalColumn)) and (0 in tmp))