[mis] fix flaky test of test_cuda_device_count_stateless (#5546)

48f589e1 · youkaichao · GitHub · 348616ac · 48f589e1
Unverified Commit 48f589e1 authored Jun 14, 2024 by youkaichao Committed by GitHub Jun 14, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/distributed/test_utils.py tests/distributed/test_utils.py +2 -1

No files found.
--- 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