Fix test_multiple_actor_reconstruction failure (#3641)

* Fix test_multiple_actor_reconstruction failure

* add comment
This commit is contained in:
Hao Chen
2018-12-27 13:57:52 -08:00
committed by Stephanie Wang
parent ac792d70c8
commit 62af2f25be
+8 -2
View File
@@ -40,13 +40,14 @@ def shutdown_only():
@pytest.fixture
def head_node_cluster():
def head_node_cluster(request):
timeout = getattr(request, 'param', 200)
cluster = ray.test.cluster_utils.Cluster(
initialize_head=True,
connect=True,
head_node_args={
"_internal_config": json.dumps({
"initial_reconstruction_timeout_milliseconds": 200,
"initial_reconstruction_timeout_milliseconds": timeout,
"num_heartbeats_timeout": 10,
})
})
@@ -2248,6 +2249,11 @@ def test_actor_reconstruction_on_node_failure(head_node_cluster):
ray.get(actor.increase.remote())
# NOTE(hchen): we set initial_reconstruction_timeout_milliseconds to 1s for
# this test. Because if this value is too small, suprious task reconstruction
# may happen and cause the test fauilure. If the value is too large, this test
# could be very slow. We can remove this once we support dynamic timeout.
@pytest.mark.parametrize('head_node_cluster', [1000], indirect=True)
def test_multiple_actor_reconstruction(head_node_cluster):
# This test can be made more stressful by increasing the numbers below.
# The total number of actors created will be