Fixup pylibcudf transition

Signed-off-by: Sebastian Berg <[email protected]>
rapidsai · Dec 5, 2024 · d5635f7 · d5635f7
1 parent d80db36
commit d5635f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/legate_dataframe/lib/core/data_type.pyx b/python/legate_dataframe/lib/core/data_type.pyx
@@ -10,7 +10,7 @@ from pylibcudf.types cimport DataType
 from pylibcudf.types cimport data_type as cpp_cudf_type
 
 import cudf
-from cudf._lib import pylibcudf
+import pylibcudf
 from cudf._lib.types import PYLIBCUDF_TO_SUPPORTED_NUMPY_TYPES, dtype_to_pylibcudf_type
 from numpy.typing import DTypeLike
 

diff --git a/python/tests/test_binaryop.py b/python/tests/test_binaryop.py
@@ -86,11 +86,11 @@ def test_scalar_input(cudf_column, op, scalar):
 
     res = binary_operation(col, scalar, op, cudf_column.dtype)
     expect = cudf_binaryop(cudf_column, cudf_scalar, op_str, cudf_column.dtype)
-    assert_frame_equal(res, cudf.Series(expect))
+    assert_frame_equal(res, expect)
 
     res = binary_operation(scalar, col, op, cudf_column.dtype)
     expect = cudf_binaryop(cudf_scalar, cudf_column, op_str, cudf_column.dtype)
-    assert_frame_equal(res, cudf.Series(expect))
+    assert_frame_equal(res, expect)
 
     with pytest.raises(ValueError, match="both inputs cannot be scalars"):
         binary_operation(scalar, scalar, op, cudf_column.dtype)