Adapt test

rapidsai · Oct 10, 2024 · a454876 · a454876
1 parent edda56c
commit a454876
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/cudf_polars/tests/test_config.py b/python/cudf_polars/tests/test_config.py
@@ -10,18 +10,18 @@
 
 import rmm
 
-from cudf_polars.dsl.ir import IR
+from cudf_polars.dsl.ir import DataFrameScan
 from cudf_polars.testing.asserts import (
     assert_gpu_result_equal,
     assert_ir_translation_raises,
 )
 
 
 def test_polars_verbose_warns(monkeypatch):
-    def raise_unimplemented(self, schema):
+    def raise_unimplemented(self, *args):
         raise NotImplementedError("We don't support this")
 
-    monkeypatch.setattr(IR, "__init__", raise_unimplemented)
+    monkeypatch.setattr(DataFrameScan, "__init__", raise_unimplemented)
     q = pl.LazyFrame({})
     # Ensure that things raise
     assert_ir_translation_raises(q, NotImplementedError)