From 98f57ee8aaca1d337d66f2a0ef25184f6f54f248 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Fri, 20 Dec 2024 22:25:40 +0100 Subject: [PATCH] Always test `pynvml` (#1421) Do not skip `pynvml` if it's not importable, given `pynvml` is a hard-dependency. Authors: - Peter Andreas Entschev (https://github.com/pentschev) Approvers: - https://github.com/jakirkham - James Lamb (https://github.com/jameslamb) URL: https://github.com/rapidsai/dask-cuda/pull/1421 --- dask_cuda/tests/test_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/dask_cuda/tests/test_utils.py b/dask_cuda/tests/test_utils.py index a0a77677d..a5e78db2a 100644 --- a/dask_cuda/tests/test_utils.py +++ b/dask_cuda/tests/test_utils.py @@ -1,6 +1,7 @@ import os from unittest.mock import patch +import pynvml import pytest from numba import cuda @@ -197,7 +198,6 @@ def test_get_ucx_config(enable_tcp_over_ucx, enable_infiniband, enable_nvlink): def test_parse_visible_devices(): - pynvml = pytest.importorskip("pynvml") pynvml.nvmlInit() indices = [] uuids = [] @@ -250,7 +250,6 @@ def test_parse_device_memory_limit(): def test_parse_visible_mig_devices(): - pynvml = pytest.importorskip("pynvml") pynvml.nvmlInit() for index in range(get_gpu_count()): handle = pynvml.nvmlDeviceGetHandleByIndex(index)