From 22859e8bd3499853e8b1e742e3872ed87c28c147 Mon Sep 17 00:00:00 2001 From: Michael Droettboom Date: Fri, 15 May 2026 23:35:07 -0700 Subject: [PATCH] Fix nvbug 5880275 --- cuda_bindings/tests/nvml/test_device.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cuda_bindings/tests/nvml/test_device.py b/cuda_bindings/tests/nvml/test_device.py index c9cb967efb8..4b5f6a1e160 100644 --- a/cuda_bindings/tests/nvml/test_device.py +++ b/cuda_bindings/tests/nvml/test_device.py @@ -4,6 +4,7 @@ from functools import cache +import numpy as np import pytest from cuda.bindings import nvml @@ -78,7 +79,7 @@ def test_get_nv_link_supported_bw_modes(all_devices): assert not hasattr(modes, "total_bw_modes") for mode in modes.bw_modes: - assert isinstance(mode, int) + assert isinstance(mode, np.uint8) def test_device_get_pdi(all_devices):