in _private test utils

Signed-off-by: dayshah <dhyey2019@gmail.com>

Author: dayshah

python/ray/_common/test_utils.py

@@ -247,12 +247,3 @@ def f():
         assert all(
             [extra_usage_tags[k] == v for k, v in expected_extra_usage_tags.items()]
         ), extra_usage_tags
-
-
-RPC_FAILURE_MAP = {
-    "request": "100:0:0",
-    "response": "0:100:0",
-    "in_flight": "0:0:100",
-}
-
-RPC_FAILURE_TYPES = list(RPC_FAILURE_MAP.keys())

python/ray/_private/test_utils.py

@@ -2103,3 +2103,12 @@ def _execute_command_on_node(command: str, node_ip: str):
     except subprocess.CalledProcessError as e:
         print("Exit code:", e.returncode)
         print("Stderr:", e.stderr)
+
+
+RPC_FAILURE_MAP = {
+    "request": "100:0:0",
+    "response": "0:100:0",
+    "in_flight": "0:0:100",
+}
+
+RPC_FAILURE_TYPES = list(RPC_FAILURE_MAP.keys())

python/ray/tests/test_actor_lineage_reconstruction.py

@@ -6,7 +6,7 @@
 import pytest
 
 import ray
-from ray._common.test_utils import (
+from ray._private.test_utils import (
     RPC_FAILURE_MAP,
     RPC_FAILURE_TYPES,
     wait_for_condition,

python/ray/tests/test_core_worker_fault_tolerance.py

@@ -4,7 +4,7 @@
 import pytest
 
 import ray
-from ray._common.test_utils import (
+from ray._private.test_utils import (
     RPC_FAILURE_MAP,
     RPC_FAILURE_TYPES,
     SignalActor,

python/ray/tests/test_object_manager_fault_tolerance.py

@@ -4,12 +4,12 @@
 import pytest
 
 import ray
-from ray._common.test_utils import (
+from ray._private.internal_api import get_memory_info_reply, get_state_from_address
+from ray._private.test_utils import (
     RPC_FAILURE_MAP,
     RPC_FAILURE_TYPES,
     wait_for_condition,
 )
-from ray._private.internal_api import get_memory_info_reply, get_state_from_address
 from ray.util.scheduling_strategies import NodeAffinitySchedulingStrategy
 
 

python/ray/tests/test_raylet_fault_tolerance.py

@@ -4,7 +4,7 @@
 import pytest
 
 import ray
-from ray._common.test_utils import (
+from ray._private.test_utils import (
     RPC_FAILURE_MAP,
     RPC_FAILURE_TYPES,
     wait_for_condition,