fix

mrocklin · Dec 22, 2023 · d5329bd · d5329bd
1 parent 143c6ba
commit d5329bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py
@@ -695,15 +695,15 @@ def test_is_dask_collection_dask_expr():
 def test_is_dask_collection_dask_expr_does_not_materialize():
     dx = pytest.importorskip("dask_expr")
 
-    class DoNotMaterialize(dx.Expr):
+    class DoNotMaterialize(dx._core.Expr):
         @property
         def _meta(self):
             return 0
 
         def __dask_graph__(self):
             assert False, "must not reach"
 
-    dyf = dx.new_collection(DoNotMaterialize())
+    dyf = dx._collection.new_collection(DoNotMaterialize())
     assert is_dask_collection(dyf)
 
     with pytest.raises(AssertionError, match="must not reach"):