From 604b140f002ce8e2f63f45253257fa8500fb5d4f Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Fri, 28 Jan 2022 10:19:04 -0600 Subject: [PATCH] Skip shuffle tests if pandas not installed --- distributed/shuffle/tests/test_graph.py | 5 +++++ distributed/shuffle/tests/test_shuffle_extension.py | 4 ++-- 2 files changed, 7 insertions(+), 2 deletions(-) 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