From d5329bd0725bc5816b0555db5e7d691263566c41 Mon Sep 17 00:00:00 2001 From: crusaderky Date: Fri, 22 Dec 2023 16:23:59 +0000 Subject: [PATCH] fix --- dask/tests/test_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py index ed62ac92205..72908f6480f 100644 --- a/dask/tests/test_base.py +++ b/dask/tests/test_base.py @@ -695,7 +695,7 @@ 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 @@ -703,7 +703,7 @@ def _meta(self): 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"):