[cli] create_or_update_cluster fix (#10085)

This commit is contained in:
Richard Liaw
2020-08-13 00:54:45 -07:00
committed by GitHub
parent 86b1db3f11
commit 7a56c3b71a
2 changed files with 21 additions and 8 deletions
+10 -6
View File
@@ -94,12 +94,16 @@ def request_resources(num_cpus=None, bundles=None):
r.publish(AUTOSCALER_RESOURCE_REQUEST_CHANNEL, json.dumps(bundles))
def create_or_update_cluster(
config_file: str, override_min_workers: Optional[int],
override_max_workers: Optional[int], no_restart: bool,
restart_only: bool, yes: bool, override_cluster_name: Optional[str],
no_config_cache: bool, dump_command_output: bool,
use_login_shells: bool) -> None:
def create_or_update_cluster(config_file: str,
override_min_workers: Optional[int],
override_max_workers: Optional[int],
no_restart: bool,
restart_only: bool,
yes: bool,
override_cluster_name: Optional[str],
no_config_cache: bool,
dump_command_output: bool = True,
use_login_shells: bool = True) -> None:
"""Create or updates an autoscaling Ray cluster from a config json."""
set_using_login_shells(use_login_shells)
cmd_output_util.set_output_redirected(not dump_command_output)