Fix importing of conftest.py

mdboom · mdboom · commit 08081396e9f9 · 2026-01-16T14:05:25.000-05:00
diff --git a/cuda_core/tests/system/conftest.py b/cuda_core/tests/system/conftest.py
@@ -14,7 +14,7 @@
 
 
 @contextmanager
-def unsupported_before(device: system.Device, expected_device_arch: system.DeviceArch | str | None):
+def unsupported_before(device, expected_device_arch):
     device_arch = device.arch
 
     if isinstance(expected_device_arch, system.DeviceArch):
