ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add AWS auto-scaler multi-line support for both extra bash script and extra trains.conf data

Browse Source
This commit is contained in:
allegroai 2020-10-12 11:07:01 +03:00
parent 0c3745ea3f
commit 994ed1a5ae
2 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/services/aws-autoscaler/aws_autoscaler.py
View File

@ -10,7 +10,7 @@ from six.moves import input
from trains import Task from trains import Task
from trains.automation.aws_auto_scaler import AwsAutoScaler from trains.automation.aws_auto_scaler import AwsAutoScaler
from trains.config import running_remotely from trains.config import running_remotely
from trains.utilities.wizard.user_input import get_input, input_int, input_bool from trains.utilities.wizard.user_input import get_input, input_int, input_bool, multiline_input
CONF_FILE = "aws_autoscaler.yaml" CONF_FILE = "aws_autoscaler.yaml"
DEFAULT_DOCKER_IMAGE = "nvidia/cuda:10.1-runtime-ubuntu18.04" DEFAULT_DOCKER_IMAGE = "nvidia/cuda:10.1-runtime-ubuntu18.04"
@ -190,9 +190,15 @@ def run_wizard():
configurations.resource_configurations = resource_configurations configurations.resource_configurations = resource_configurations
configurations.extra_vm_bash_script = input( configurations.extra_vm_bash_script, num_lines_bash_script = multiline_input(
"\nEnter any pre-execution bash script to be executed on the newly created instances []: " "\nEnter any pre-execution bash script to be executed on the newly created instances []"
) )
print("Entered {} lines of pre-execution bash script".format(num_lines_bash_script))
configurations.extra_trains_conf, num_lines_trains_conf = multiline_input(
"\nEnter anything you'd like to include in your trains.conf file []"
)
print("Entered {} extra lines for trains.conf file".format(num_lines_trains_conf))
print("\nDefine the machines budget:") print("\nDefine the machines budget:")
print("-----------------------------") print("-----------------------------")

12
trains/utilities/wizard/user_input.py
View File

@ -69,3 +69,15 @@ def input_bool(question, default=False):
raise ValueError() raise ValueError()
except ValueError: except ValueError:
print("Invalid input: please enter 'yes' or 'no'") print("Invalid input: please enter 'yes' or 'no'")
def multiline_input(description=""):
print("{} \nNote: two consecutive empty lines would terminate the input : ".format(description))
lines = []
empty_lines = 0
while empty_lines < 2:
line = input()
lines.append(line)
empty_lines = 0 if line else empty_lines + 1
res = "\n".join(lines[:-1])
return res, len(res.splitlines())