finito

main.py

@@ -35,20 +35,20 @@ def main():
     print(f"\n--- Evaluating Initial Architecture ---")
     red_team_agent.attack_program.eval_program(attacker_prog, trainset)
 
-    optimizer = MIPROv2(metric=metric, auto=None)
+    optimizer = MIPROv2(metric=red_team_agent.attack_program.metric, auto=None)
     best_prog = optimizer.compile(
         attacker_prog,
         trainset=trainset,
         max_bootstrapped_demos=2,
         max_labeled_demos=0,
         num_trials=3,
+        num_candidates=6,
     )
 
     # evaluating architecture DSPy post-compilation
     print(f"\n--- Evaluating Optimized Architecture ---")
     red_team_agent.attack_program.eval_program(best_prog, trainset)
     """
-
     # push to hub
     red_team_agent.push_to_hub("farouk1/redteam", commit_message="finito", with_code=True)
     print("---------Pushed to hub!---------")