update import

rapidsai · Dec 11, 2024 · 453f3cf · 453f3cf
1 parent 6a0483b
commit 453f3cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py
@@ -39,7 +39,7 @@
 import cudf  # noqa: E402
 import rmm  # noqa: E402
 from rmm.allocators.cupy import rmm_cupy_allocator  # noqa: E402
-
+from cudf.core.abc import Serializable
 
 def sizeof_cudf_dataframe(df):
     return int(
@@ -72,7 +72,7 @@ async def recv_df(ep):
     for frame in frames:
         await ep.recv(frame)
 
-    return cudf.Serializable.device_deserialize(header, frames)
+    return Serializable.device_deserialize(header, frames)
 
 
 async def barrier(rank, eps):