[mis] fix flaky test of test_cuda_device_count_stateless (vllm-projec…

…t#5546)
xjpang · Jul 24, 2024 · f216f46 · f216f46
1 parent baa74f1
commit f216f46
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/distributed/test_utils.py b/tests/distributed/test_utils.py
@@ -23,7 +23,8 @@ def test_cuda_device_count_stateless():
     CUDA_VISIBLE_DEVICES is changed."""
 
     actor = _CUDADeviceCountStatelessTestActor.options(num_gpus=2).remote()
-    assert ray.get(actor.get_cuda_visible_devices.remote()) == "0,1"
+    assert sorted(ray.get(
+        actor.get_cuda_visible_devices.remote()).split(",")) == ["0", "1"]
     assert ray.get(actor.get_count.remote()) == 2
     ray.get(actor.set_cuda_visible_devices.remote("0"))
     assert ray.get(actor.get_count.remote()) == 1