Skip to content

Commit 54d72cd

Browse files
authored
Fix test_deprecation_warning in gpuCI (dask-contrib#248)
* Try to assess wher extra warning is coming from * Use assertion to print out warning messages * Filter out non deprecation warnings from deprecation test * Try passing gpu argument directly to register_dask_table
1 parent 5ab6d98 commit 54d72cd

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/unit/test_context.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,10 @@ def test_deprecation_warning(gpu):
4040
c = Context()
4141
data_frame = dd.from_pandas(pd.DataFrame(), npartitions=1)
4242

43-
if gpu:
44-
data_frame = dask_cudf.from_dask_dataframe(data_frame)
45-
4643
with warnings.catch_warnings(record=True) as w:
4744
warnings.simplefilter("always")
4845

49-
c.register_dask_table(data_frame, "table")
46+
c.register_dask_table(data_frame, "table", gpu=gpu)
5047

5148
assert len(w) == 1
5249
assert issubclass(w[-1].category, DeprecationWarning)

0 commit comments

Comments
 (0)