mirror of
https://github.com/wassname/ray.git
synced 2026-07-15 11:25:40 +08:00
Fix test_multiple_actor_reconstruction failure (#3641)
* Fix test_multiple_actor_reconstruction failure * add comment
This commit is contained in:
+8
-2
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user