Skip to content

Commit

Permalink
Relax type-check in test_proxy.py (#1321)
Browse files Browse the repository at this point in the history
There is a minor [missing feature in dask-expr](dask/dask-expr#985). I suggest we relax this test a bit to handle both "legacy" and "new" DataFrame collections.

Authors:
  - Richard (Rick) Zamora (https://github.com/rjzamora)

Approvers:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

URL: #1321
  • Loading branch information
rjzamora authored Mar 18, 2024
1 parent 6c21736 commit 62c27d5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions dask_cuda/tests/test_proxy.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 62c27d5

Please sign in to comment.