We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 82c5f98 commit c08bc32Copy full SHA for c08bc32
2 files changed
cuda_core/cuda/core/system/typing.py
@@ -362,3 +362,6 @@ class DeviceArch(_FastEnum):
362
363
except ImportError:
364
pass
365
+
366
367
+del StrEnum, sys
cuda_core/tests/system/test_system_device.py
@@ -500,7 +500,7 @@ def test_get_nearest_gpus():
500
501
for device in system.Device.get_all_devices():
502
for near_device in device.get_topology_nearest_gpus(typing.GpuTopologyLevel.SINGLE):
503
- assert isinstance(near_device, typing.Device)
+ assert isinstance(near_device, system.Device)
504
505
506
@pytest.mark.skipif(helpers.IS_WSL or helpers.IS_WINDOWS, reason="Device attributes not supported on WSL or Windows")
0 commit comments