Using pytest.importorskip("cupy")

rapidsai · Nov 11, 2019 · fcc8add · fcc8add
1 parent 9c0bf1b
commit fcc8add
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dask_cuda/tests/test_device_host_file.py b/dask_cuda/tests/test_device_host_file.py
@@ -1,11 +1,11 @@
 import numpy as np
-import cupy
 from dask_cuda.device_host_file import DeviceHostFile, host_to_device, device_to_host
 from distributed.protocol import deserialize_bytes, serialize_bytelist
 from random import randint
 import dask.array as da
 
 import pytest
+cupy = pytest.importorskip("cupy")
 
 
 @pytest.mark.xfail(reason="https://github.com/rapidsai/dask-cuda/pull/171")

diff --git a/dask_cuda/tests/test_dgx.py b/dask_cuda/tests/test_dgx.py
@@ -3,7 +3,6 @@
 import psutil
 import os
 
-import cupy
 import dask.array as da
 
 from distributed.utils_test import gen_test
@@ -13,6 +12,7 @@
 from dask_cuda.initialize import initialize
 from dask_cuda.utils import get_ucx_env
 
+cupy = pytest.importorskip("cupy")
 
 @pytest.mark.skipif(
     "ib0" not in psutil.net_if_addrs(), reason="Infiniband interface ib0 not found"