[https://nvbugs/6029864][fix] Fix flaky ray issue

brb-nv · brb-nv · commit 1fb85a4646b0 · 2026-04-02T16:19:50.000Z
Signed-off-by: Balaram Buddharaju &lt;169953907+brb-nv@users.noreply.github.com&gt;
diff --git a/tests/unittest/conftest.py b/tests/unittest/conftest.py
@@ -433,6 +433,8 @@ def process_gpu_memory_info_available():
 
 @pytest.fixture(scope="function")
 def setup_ray_cluster() -> Generator[int, None, None]:
+    import time
+
     runtime_env = {
         "env_vars": {
             "RAY_EXPERIMENTAL_NOSET_CUDA_VISIBLE_DEVICES": "1"
@@ -448,6 +450,8 @@ def setup_ray_cluster() -> Generator[int, None, None]:
         ray.init(address="local", **ray_init_args)
         gcs_addr = ray.get_runtime_context().gcs_address
         port = int(gcs_addr.split(":")[1])
+        # Allow raylet to complete GCS registration before tests create actors
+        time.sleep(2)
         yield port
     finally:
         if ray.is_initialized():
