Skip to content

Commit c08bc32

Browse files
committed
Fix del and fix a test
1 parent 82c5f98 commit c08bc32

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

cuda_core/cuda/core/system/typing.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,3 +362,6 @@ class DeviceArch(_FastEnum):
362362

363363
except ImportError:
364364
pass
365+
366+
367+
del StrEnum, sys

cuda_core/tests/system/test_system_device.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ def test_get_nearest_gpus():
500500

501501
for device in system.Device.get_all_devices():
502502
for near_device in device.get_topology_nearest_gpus(typing.GpuTopologyLevel.SINGLE):
503-
assert isinstance(near_device, typing.Device)
503+
assert isinstance(near_device, system.Device)
504504

505505

506506
@pytest.mark.skipif(helpers.IS_WSL or helpers.IS_WINDOWS, reason="Device attributes not supported on WSL or Windows")

0 commit comments

Comments
 (0)