From 890cb8cb70f095c146e747db2c3368f6af1931fa Mon Sep 17 00:00:00 2001 From: SangBin Cho Date: Wed, 10 Jun 2020 14:21:27 -0700 Subject: [PATCH] Node failure test fix (#8882) --- ci/long_running_tests/workloads/node_failures.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ci/long_running_tests/workloads/node_failures.py b/ci/long_running_tests/workloads/node_failures.py index 204a6aa6f..e6f179bbe 100644 --- a/ci/long_running_tests/workloads/node_failures.py +++ b/ci/long_running_tests/workloads/node_failures.py @@ -4,7 +4,7 @@ import time import ray from ray.cluster_utils import Cluster -from ray.test_utils import get_non_head_nodes +from ray.test_utils import get_other_nodes num_redis_shards = 5 redis_max_memory = 10**8 @@ -51,8 +51,8 @@ while True: for _ in range(100): previous_ids = [f.remote(previous_id) for previous_id in previous_ids] + node_to_kill = get_other_nodes(cluster, exclude_head=True)[0] - node_to_kill = get_non_head_nodes(cluster)[0] # Remove the first non-head node. cluster.remove_node(node_to_kill) cluster.add_node()