def main()

in cli.py [0:0]


def main():
    args = parse_args()

    # Configure logging
    log_level = logging.DEBUG if args.verbose else logging.INFO
    logging.basicConfig(
        format="%(asctime)s - %(message)s",
        datefmt="%Y-%m-%d %H:%M:%S",
        level=log_level
    )

    # Initialize Trending with provided arguments
    trending = Trending(
        tasks=args.tasks,
        max_models_per_task=args.max_models_per_task,
        budget=args.budget
    )

    # Run the trending model selection and deployment
    selected_models, max_reward, spent_budget = trending(filename=args.filename, deploy_models=not args.dry)

    logging.info(f"Selected {len(selected_models)} models with total reward of {max_reward}")
    logging.info(f"Spent budget: ${spent_budget:,} out of ${args.budget:,}")