From e5954693bc63b5df8548020b2717e62ef2cdf403 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Thu, 19 Dec 2024 08:34:39 -0800 Subject: [PATCH] Replace as_device_scalar with cudf.Scalar.device_value Signed-off-by: Sebastian Berg --- python/legate_dataframe/lib/core/scalar.pyx | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/legate_dataframe/lib/core/scalar.pyx b/python/legate_dataframe/lib/core/scalar.pyx index 863ff27..8e24292 100644 --- a/python/legate_dataframe/lib/core/scalar.pyx +++ b/python/legate_dataframe/lib/core/scalar.pyx @@ -13,7 +13,6 @@ import numbers import cudf import legate.core import numpy -from cudf._lib.scalar import as_device_scalar ScalarLike = numpy.number | numbers.Number | cudf.Scalar | legate.core.Scalar @@ -39,5 +38,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 = as_device_scalar(scalar) + cdef DeviceScalar cudf_scalar = (cudf.Scalar(scalar).device_value) return cpp_ScalarArg(dereference(cudf_scalar.get_raw_ptr()))