diff --git a/distributed/shuffle/tests/test_graph.py b/distributed/shuffle/tests/test_graph.py index 256f9132cd4..42d6be98d3b 100644 --- a/distributed/shuffle/tests/test_graph.py +++ b/distributed/shuffle/tests/test_graph.py @@ -3,6 +3,11 @@ import asyncio from typing import TYPE_CHECKING +import pytest + +pd = pytest.importorskip("pandas") +pytest.importorskip("dask.dataframe") + import dask import dask.dataframe as dd from dask.blockwise import Blockwise diff --git a/distributed/shuffle/tests/test_shuffle_extension.py b/distributed/shuffle/tests/test_shuffle_extension.py index f47c62e9ca7..ad536280672 100644 --- a/distributed/shuffle/tests/test_shuffle_extension.py +++ b/distributed/shuffle/tests/test_shuffle_extension.py @@ -5,10 +5,10 @@ from collections import Counter from typing import TYPE_CHECKING -import pandas as pd import pytest -import dask.dataframe as dd +pd = pytest.importorskip("pandas") +dd = pytest.importorskip("dask.dataframe") from distributed.utils_test import gen_cluster