in cli.py [0:0]
def parse_args():
parser = argparse.ArgumentParser(
description="Deploy trending models based on optimization criteria"
)
parser.add_argument(
"--tasks",
type=str,
nargs="+",
help="List of tasks to consider. If not provided, defaults to predefined tasks."
)
parser.add_argument(
"--max-models-per-task",
type=int,
default=300,
help="Maximum number of models to consider per task. Default: 200"
)
parser.add_argument(
"--budget",
type=int,
default=10_000,
help="Budget for model deployment in monthly dollar spend. Default: 1000"
)
parser.add_argument(
"--filename",
type=str,
help="Path to save selected models as JSON. If not provided, models won't be saved to file."
)
parser.add_argument(
"--dry",
action="store_true",
help="Run in dry run mode. No models will be deployed."
)
parser.add_argument(
"--verbose",
action="store_true",
help="Enable verbose output"
)
return parser.parse_args()