Fix imports

mdboom · mdboom · commit b97f3c8d1ea7 · 2026-01-20T10:41:38.000-05:00
diff --git a/cuda_bindings/tests/nvml/conftest.py b/cuda_bindings/tests/nvml/conftest.py
@@ -5,6 +5,7 @@
 
 import pytest
 from cuda.bindings import _nvml as nvml
+from cuda.bindings._test_helpers.arch_check import unsupported_before  # noqa: F401
 
 
 class NVMLInitializer:
