97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
import click
|
|
import json
|
|
|
|
from kfp.cli.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.list_experiments(
|
|
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.delete_experiment(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)
|