[tune] Add --columns flag for CLI (#4564)

This commit is contained in:
Andrew Tan
2019-04-05 19:49:01 -07:00
committed by Richard Liaw
parent 300ec72d15
commit 991b911e1d
4 changed files with 51 additions and 10 deletions
+28 -4
View File
@@ -28,9 +28,26 @@ def cli():
default=None,
type=str,
help="Select filter in the format '<column> <operator> <value>'.")
def list_trials(experiment_path, sort, output, filter_op):
@click.option(
"--columns",
default=None,
type=str,
help="Select columns to be displayed.")
@click.option(
"--result-columns",
"result_columns",
default=None,
type=str,
help="Select columns of last result to be displayed.")
def list_trials(experiment_path, sort, output, filter_op, columns,
result_columns):
"""Lists trials in the directory subtree starting at the given path."""
commands.list_trials(experiment_path, sort, output, filter_op)
if columns:
columns = columns.split(',')
if result_columns:
result_columns = result_columns.split(',')
commands.list_trials(experiment_path, sort, output, filter_op, columns,
result_columns)
@cli.command()
@@ -50,9 +67,16 @@ def list_trials(experiment_path, sort, output, filter_op):
default=None,
type=str,
help="Select filter in the format '<column> <operator> <value>'.")
def list_experiments(project_path, sort, output, filter_op):
@click.option(
"--columns",
default=None,
type=str,
help="Select columns to be displayed.")
def list_experiments(project_path, sort, output, filter_op, columns):
"""Lists experiments in the directory subtree."""
commands.list_experiments(project_path, sort, output, filter_op)
if columns:
columns = columns.split(',')
commands.list_experiments(project_path, sort, output, filter_op, columns)
@cli.command()