Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import nemo_evaluator_launcher.cli.kill as kill
import nemo_evaluator_launcher.cli.ls_runs as ls_runs
import nemo_evaluator_launcher.cli.ls_tasks as ls_tasks
import nemo_evaluator_launcher.cli.quickstart as quickstart
import nemo_evaluator_launcher.cli.run as run
import nemo_evaluator_launcher.cli.status as status
import nemo_evaluator_launcher.cli.version as version
Expand All @@ -41,6 +42,7 @@ def is_verbose_enabled(args) -> bool:
# Check subcommand verbose flags
subcommands = [
"run",
"quickstart",
"status",
"info",
"kill",
Expand Down Expand Up @@ -94,6 +96,14 @@ def create_parser() -> ArgumentParser:
)
run_parser.add_arguments(run.Cmd, dest="run")

# Quickstart subcommand
quickstart_parser = subparsers.add_parser(
"quickstart",
help="Interactive quickstart to generate a config",
description="Interactive questionnaire that generates a YAML config referencing internal defaults",
)
quickstart_parser.add_arguments(quickstart.Cmd, dest="quickstart")

# Status subcommand
status_parser = subparsers.add_parser(
"status", help="Check job status", description="Check job status"
Expand Down Expand Up @@ -202,6 +212,8 @@ def main() -> None:
args.version.execute()
elif args.command == "run":
args.run.execute()
elif args.command == "quickstart":
args.quickstart.execute()
elif args.command == "status":
args.status.execute()
elif args.command == "kill":
Expand Down
Loading