Set RAY_FORCE_DIRECT=1 for run_rllib_tests, test_basic (#6171)

This commit is contained in:
Eric Liang
2019-11-25 14:12:11 -08:00
committed by GitHub
parent c9314098b9
commit 64a3a7239e
16 changed files with 229 additions and 123 deletions
+8
View File
@@ -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