Skip to content

Add Setup CLI#222

Merged
AnirudhDagar merged 6 commits into
autogluon:masterfrom
AnirudhDagar:add_cli
May 27, 2026
Merged

Add Setup CLI#222
AnirudhDagar merged 6 commits into
autogluon:masterfrom
AnirudhDagar:add_cli

Conversation

@AnirudhDagar
Copy link
Copy Markdown
Collaborator

Description of changes:
Following up #213

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Click-based wrapper around the four cloud_setup functions (bootstrap,
register, status, teardown). The Python API stays untouched; this is a
thin interactive layer with Rich prompts, deployment-plan preview,
destructive-action confirmation, and clean RuntimeError → exit code
translation.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Comment thread src/autogluon/cloud/cli.py
Comment thread src/autogluon/cloud/cli.py Outdated
table.add_row("Stack", report.config.stack_name)
table.add_section()
for check_name, result in report.checks.items():
color = "green" if result.startswith("ok") else "red"
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Stack status CREATE_COMPLETE is colored red - this looks wrong

default_stack = f"ag-cloud-{backend.replace('_', '-')}"
stack_name = Prompt.ask("Stack name", default=default_stack)
effective_stack = stack_name
template_url = f"https://github.com/autogluon/autogluon-cloud/blob/master/cloudformation/ag_cloud_{backend}.yaml"
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let's change this to the URL pointing to the correct stable branch (e.g. 0.5.0) in the follow-up PR where we also get rid of the copy-on-setup logic for CFN templates

Comment thread src/autogluon/cloud/cli.py Outdated


@cli.command()
@click.option("--backend", type=click.Choice(SUPPORTED_BACKENDS), default=None, help="Which backend to use.")
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let's hide the ray_aws backend for now? We are not properly testing it and we don't want to highlight it in the next release IMO

- Replace SUPPORTED_BACKENDS with local _CLI_BACKENDS = ("sagemaker",);
  ray_aws is hidden from the CLI until it is properly tested
- bootstrap: --backend now defaults to "sagemaker" (no prompt)
- status: paint *_COMPLETE stack states green, not red
- Trim verbose paragraph from module docstring

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
@AnirudhDagar AnirudhDagar marked this pull request as ready for review May 27, 2026 10:53
@AnirudhDagar AnirudhDagar changed the title [WIP] Add Setup CLI Add Setup CLI May 27, 2026
Copy link
Copy Markdown
Collaborator

@shchur shchur left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@AnirudhDagar AnirudhDagar merged commit e901fc5 into autogluon:master May 27, 2026
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants