diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py index 9977a185d..fc576058f 100644 --- a/dask_cuda/tests/test_proxy.py +++ b/dask_cuda/tests/test_proxy.py @@ -1,6 +1,7 @@ import operator import pickle +import pandas import pytest from pandas.testing import assert_frame_equal @@ -222,6 +223,10 @@ def task(x): ddf = dask_cudf.from_cudf(df, npartitions=1) ddf = ddf.map_partitions(task, meta=df.head()) got = ddf.compute() + if isinstance(got, pandas.Series): + pytest.xfail( + "BUG fixed by " + ) assert_frame_equal(got.to_pandas(), df.to_pandas())