From 85c64d4bdedfbeebc2e278f1e0b31a343b0f9666 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Mon, 18 Mar 2024 10:21:35 -0700 Subject: [PATCH] relax type-check --- dask_cuda/tests/test_proxy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py index 5458c5ba..31a9e996 100644 --- a/dask_cuda/tests/test_proxy.py +++ b/dask_cuda/tests/test_proxy.py @@ -537,10 +537,10 @@ def test_from_cudf_of_proxy_object(): assert has_parallel_type(df) ddf = dask_cudf.from_cudf(df, npartitions=1) - assert has_parallel_type(ddf) + assert has_parallel_type(ddf._meta) # Notice, the output is a dask-cudf dataframe and not a proxy object - assert type(ddf) is dask_cudf.core.DataFrame + assert type(ddf._meta) is cudf.DataFrame def test_proxy_object_parquet(tmp_path):