From f6717b8b0383580a534a6b87135830a350cf619a Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Wed, 4 Nov 2020 22:40:07 -0800 Subject: [PATCH] [autoscaler] Support empty node list for kill node (#11810) --- python/ray/autoscaler/_private/commands.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ray/autoscaler/_private/commands.py b/python/ray/autoscaler/_private/commands.py index dd42009f8..df0bd4aa3 100644 --- a/python/ray/autoscaler/_private/commands.py +++ b/python/ray/autoscaler/_private/commands.py @@ -384,7 +384,7 @@ def teardown_cluster(config_file: str, yes: bool, workers_only: bool, def kill_node(config_file: str, yes: bool, hard: bool, - override_cluster_name: Optional[str]) -> str: + override_cluster_name: Optional[str]) -> Optional[str]: """Kills a random Raylet worker.""" config = yaml.safe_load(open(config_file).read()) @@ -398,6 +398,9 @@ def kill_node(config_file: str, yes: bool, hard: bool, nodes = provider.non_terminated_nodes({ TAG_RAY_NODE_KIND: NODE_KIND_WORKER }) + if not nodes: + cli_logger.print("No worker nodes detected.") + return None node = random.choice(nodes) cli_logger.print("Shutdown " + cf.bold("{}"), node) if hard: