109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
import click
|
|
import json
|
|
|
|
from .output import print_output, OutputFormat
|
|
|
|
|
|
@click.group()
|
|
def experiment():
|
|
"""Manage experiment resources"""
|
|
pass
|
|
|
|
|
|
@experiment.command()
|
|
@click.option(
|
|
'-d',
|
|
'--description',
|
|
help="Description of the experiment."
|
|
)
|
|
@click.argument("name")
|
|
@click.pass_context
|
|
def create(ctx, description, name):
|
|
"""Create an experiment"""
|
|
client = ctx.obj["client"]
|
|
output_format = ctx.obj["output"]
|
|
|
|
response = client.create_experiment(name, description=description)
|
|
_display_experiment(response, output_format)
|
|
|
|
|
|
@experiment.command()
|
|
@click.option(
|
|
'-m',
|
|
'--max-size',
|
|
default=100,
|
|
help="Max size of the listed experiments."
|
|
)
|
|
@click.pass_context
|
|
def list(ctx, max_size):
|
|
"""List experiments"""
|
|
client = ctx.obj['client']
|
|
output_format = ctx.obj['output']
|
|
|
|
response = client.experiments.list_experiment(
|
|
page_size=max_size,
|
|
sort_by="created_at desc"
|
|
)
|
|
if response.experiments:
|
|
_display_experiments(response.experiments, output_format)
|
|
else:
|
|
if output_format == OutputFormat.json.name:
|
|
msg = json.dumps([])
|
|
else:
|
|
msg = "No experiments found"
|
|
click.echo(msg)
|
|
|
|
|
|
@experiment.command()
|
|
@click.argument("experiment-id")
|
|
@click.pass_context
|
|
def get(ctx, experiment_id):
|
|
"""Get detailed information about an experiment"""
|
|
client = ctx.obj["client"]
|
|
output_format = ctx.obj["output"]
|
|
|
|
response = client.get_experiment(experiment_id)
|
|
_display_experiment(response, output_format)
|
|
|
|
|
|
@experiment.command()
|
|
@click.argument("experiment-id")
|
|
@click.pass_context
|
|
def delete(ctx, experiment_id):
|
|
"""Delete an experiment"""
|
|
|
|
confirmation = "Caution. The RunDetails page could have an issue" \
|
|
" when it renders a run that has no experiment." \
|
|
" Do you want to continue?"
|
|
if not click.confirm(confirmation):
|
|
return
|
|
|
|
client = ctx.obj["client"]
|
|
|
|
client.experiments.delete_experiment(id=experiment_id)
|
|
click.echo("{} is deleted.".format(experiment_id))
|
|
|
|
|
|
def _display_experiments(experiments, output_format):
|
|
headers = ["Experiment ID", "Name", "Created at"]
|
|
data = [[
|
|
exp.id,
|
|
exp.name,
|
|
exp.created_at.isoformat()
|
|
] for exp in experiments]
|
|
print_output(data, headers, output_format, table_format="grid")
|
|
|
|
|
|
def _display_experiment(exp, output_format):
|
|
table = [
|
|
["ID", exp.id],
|
|
["Name", exp.name],
|
|
["Description", exp.description],
|
|
["Created at", exp.created_at.isoformat()],
|
|
]
|
|
if output_format == OutputFormat.table.name:
|
|
print_output([], ["Experiment Details"], output_format)
|
|
print_output(table, [], output_format, table_format="plain")
|
|
elif output_format == OutputFormat.json.name:
|
|
print_output(dict(table), [], output_format)
|