ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix user input wizard, support python 3.5 / 2.7

Browse Source
This commit is contained in:
allegroai 2020-07-30 15:08:27 +03:00
parent 23394a265d
commit 0a5c10b4b0
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,3 @@
import distutils
from typing import Optional from typing import Optional
@ -53,11 +52,20 @@ def input_int(
def input_bool(question, default=False): def input_bool(question, default=False):
# type: (str, bool) -> bool # type: (str, bool) -> bool
"""
:param question: string to display
:param default: default boolean value
:return: return True if response is 'y'/'yes' 't'/'true' in input.lower()
"""
while True: while True:
try: try:
response = input("{}: ".format(question)).lower() response = input("{}: ".format(question)).lower()
if not response: if not response:
return default return default
return distutils.util.strtobool(response) if response.startswith("y") or response.startswith("t"):
return True
if response.startswith("n") or response.startswith("f"):
return False
raise ValueError()
except ValueError: except ValueError:
print("Invalid input: please enter yes or no") print("Invalid input: please enter 'yes' or 'no'")