diff --git a/ucp/benchmarks/cudf_merge.py b/ucp/benchmarks/cudf_merge.py index 78164da5..fb30da41 100644 --- a/ucp/benchmarks/cudf_merge.py +++ b/ucp/benchmarks/cudf_merge.py @@ -38,8 +38,9 @@ import cudf # noqa: E402 import rmm # noqa: E402 -from rmm.allocators.cupy import rmm_cupy_allocator # noqa: E402 from cudf.core.abc import Serializable +from rmm.allocators.cupy import rmm_cupy_allocator # noqa: E402 + def sizeof_cudf_dataframe(df): return int(