diff --git a/gradient/cli/notebooks.py b/gradient/cli/notebooks.py index d50a82b1..254ef309 100644 --- a/gradient/cli/notebooks.py +++ b/gradient/cli/notebooks.py @@ -25,13 +25,6 @@ def notebook_metrics(): @notebooks_group.command("create", help="Create new notebook") -@click.option( - "--clusterId", - "cluster_id", - type=str, - help="Cluster ID", - cls=common.GradientOption, -) @click.option( "--machineType", "machine_type", @@ -40,20 +33,21 @@ def notebook_metrics(): help="Virtual Machine type label e.g. P5000", cls=common.GradientOption, ) -@click.option( - "--containerId", - "container_id", - type=int, - help="Container ID", - cls=common.GradientOption, -) @click.option( "--container", "container", + required=True, type=str, help="Container name", cls=common.GradientOption, ) +@click.option( + "--clusterId", + "cluster_id", + type=str, + help="Cluster ID", + cls=common.GradientOption, +) @click.option( "--name", "name", @@ -427,6 +421,7 @@ def artifacts(): @click.option( "--id", "notebook_id", + required=True, cls=common.GradientOption, help="ID of the notebook", )