From a1156754e9cca9671f69ff437bdf3ceb1c2cef31 Mon Sep 17 00:00:00 2001 From: Hao Chen Date: Thu, 27 Jun 2019 14:37:00 +0800 Subject: [PATCH] Fix test_task_forward (#5040) --- .../test_perf_integration.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/python/ray/tests/perf_integration_tests/test_perf_integration.py b/python/ray/tests/perf_integration_tests/test_perf_integration.py index ff34fe412..393f191e1 100644 --- a/python/ray/tests/perf_integration_tests/test_perf_integration.py +++ b/python/ray/tests/perf_integration_tests/test_perf_integration.py @@ -53,9 +53,17 @@ def benchmark_task_forward(f, num_tasks): "num_tasks", [10**3, 10**4], ids=[str(num) + "_tasks" for num in [10**3, 10**4]]) def test_task_forward(benchmark, num_tasks): - with _ray_start_cluster(num_cpus=16, object_store_memory=10**6) as cluster: - cluster.add_node(resources={"my_resource": 100}) - ray.init(redis_address=cluster.redis_address) + with _ray_start_cluster( + do_init=True, + num_nodes=1, + num_cpus=16, + object_store_memory=10**7, + ) as cluster: + cluster.add_node( + num_cpus=16, + object_store_memory=10**7, + resources={"my_resource": 100}, + ) @ray.remote(resources={"my_resource": 0.001}) def f():