mirror of
https://github.com/wassname/ray.git
synced 2026-07-04 06:44:03 +08:00
Set RAY_FORCE_DIRECT=1 for run_rllib_tests, test_basic (#6171)
This commit is contained in:
@@ -18,6 +18,8 @@ import ray.test_utils
|
||||
import ray.cluster_utils
|
||||
from ray.test_utils import run_string_as_driver
|
||||
|
||||
RAY_FORCE_DIRECT = bool(os.environ.get("RAY_FORCE_DIRECT"))
|
||||
|
||||
|
||||
def test_actor_init_error_propagated(ray_start_regular):
|
||||
@ray.remote
|
||||
@@ -807,6 +809,7 @@ def test_exception_raised_when_actor_node_dies(ray_start_cluster_head):
|
||||
@pytest.mark.skipif(
|
||||
os.environ.get("RAY_USE_NEW_GCS") == "on",
|
||||
reason="Hanging with new GCS API.")
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="no ft yet")
|
||||
def test_actor_init_fails(ray_start_cluster_head):
|
||||
cluster = ray_start_cluster_head
|
||||
remote_node = cluster.add_node()
|
||||
@@ -832,6 +835,7 @@ def test_actor_init_fails(ray_start_cluster_head):
|
||||
assert results == [1 for actor in actors]
|
||||
|
||||
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="no ft yet")
|
||||
def test_reconstruction_suppression(ray_start_cluster_head):
|
||||
cluster = ray_start_cluster_head
|
||||
num_nodes = 5
|
||||
@@ -1148,6 +1152,7 @@ def setup_queue_actor():
|
||||
ray.shutdown()
|
||||
|
||||
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="TODO support block/unblock")
|
||||
def test_fork(setup_queue_actor):
|
||||
queue = setup_queue_actor
|
||||
|
||||
@@ -1166,6 +1171,7 @@ def test_fork(setup_queue_actor):
|
||||
assert filtered_items == list(range(1))
|
||||
|
||||
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="TODO support block/unblock")
|
||||
def test_fork_consistency(setup_queue_actor):
|
||||
queue = setup_queue_actor
|
||||
|
||||
@@ -1197,6 +1203,7 @@ def test_fork_consistency(setup_queue_actor):
|
||||
assert filtered_items == list(range(num_items_per_fork))
|
||||
|
||||
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="TODO support block/unblock")
|
||||
def test_pickled_handle_consistency(setup_queue_actor):
|
||||
queue = setup_queue_actor
|
||||
|
||||
@@ -1230,6 +1237,7 @@ def test_pickled_handle_consistency(setup_queue_actor):
|
||||
assert filtered_items == list(range(num_items_per_fork))
|
||||
|
||||
|
||||
@pytest.mark.skipif(RAY_FORCE_DIRECT, reason="TODO support block/unblock")
|
||||
def test_nested_fork(setup_queue_actor):
|
||||
queue = setup_queue_actor
|
||||
|
||||
|
||||
Reference in New Issue
Block a user