Replace as_device_scalar with cudf.Scalar.device_value

rapidsai · Dec 19, 2024 · d336eff · d336eff
1 parent d43d878
commit d336eff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/legate_dataframe/lib/core/scalar.pyx b/python/legate_dataframe/lib/core/scalar.pyx
@@ -39,5 +39,5 @@ cdef cpp_ScalarArg cpp_scalar_arg_from_python(scalar: ScalarLike):
 
     # NOTE: Converting to a cudf scalar isn't really ideal, as we copy
     #       to the device, just to copy it back again to get a legate one.
-    cdef DeviceScalar cudf_scalar = <DeviceScalar>as_device_scalar(scalar)
+    cdef DeviceScalar cudf_scalar = <DeviceScalar>(cudf.Scalar(scalar).device_value)
     return cpp_ScalarArg(dereference(cudf_scalar.get_raw_ptr()))