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:,}")