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():