add run options to snapshot

main.py

@@ -249,7 +249,16 @@ def run_episode(evo_alg_str: str, debug: bool = False):
         # store new generation
         P.append([prompt.id for prompt in population])
 
-    save_snapshot(all_prompts, family_tree, P, T, N, task, evolution_model)
+    save_snapshot(
+        all_prompts,
+        family_tree,
+        P,
+        T,
+        N,
+        task,
+        evolution_model,
+        options.__dict__,
+    )
     # Line 8: Return the best prompt, p∗, among the final population PT :
     # p∗ ← argmaxp∈PT f(p, D)
     p = all_prompts[max(P[-1], key=lambda prompt_id: all_prompts[prompt_id].score)]

utils.py

@@ -129,6 +129,7 @@ def save_snapshot(
     N: int,
     task,
     model: Llama2 | OpenAI,
+    run_options: dict[str, Any],
 ):
     import json
 
@@ -147,6 +148,7 @@ def save_snapshot(
                     "metric": task.metric_name,
                 },
                 "model": {"name": model.__class__.__name__},
+                "run_options": run_options,
             },
             f,
             indent=4,