From cc0332f982b9baeb926402e4272c4bd85d62d4e1 Mon Sep 17 00:00:00 2001 From: Julia Signell Date: Mon, 8 Nov 2021 17:41:10 -0500 Subject: [PATCH] Import graphviz.Diagraph from top of module, not from dot (#8363) --- dask/tests/test_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py index 1faf821531d..1e8e8b840b3 100644 --- a/dask/tests/test_base.py +++ b/dask/tests/test_base.py @@ -1018,7 +1018,7 @@ def test_visualize_highlevelgraph(): x = da.arange(5, chunks=2) viz = x.dask.visualize(filename=os.path.join(d, "mydask.png")) # check visualization will automatically render in the jupyter notebook - assert isinstance(viz, graphviz.dot.Digraph) + assert isinstance(viz, graphviz.Digraph) @pytest.mark.skipif(