Fix key error bug

JasonGross · Nov 6, 2024 · 4ab2a73 · 4ab2a73
1 parent ec9f859
commit 4ab2a73
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks_jason/max_of_K_all_models.py b/notebooks_jason/max_of_K_all_models.py
@@ -3266,6 +3266,7 @@ def do_linear_regression(X, Y):
         ]
         print(f"len after: {len(df)}")
 
+    sorted_known_seeds = tuple(sorted(df["seed"].values))
     # Group by 'attention_error_handling' and calculate the max 'normalized-accuracy-bound' for sorting groups
     df = df[
         [
@@ -3329,7 +3330,7 @@ def do_linear_regression(X, Y):
                         sing_upper_bound,
                         attn_err_handling_key,
                         best_bound_only,
-                        tuple(sorted(subgroup["seed"].values)),
+                        sorted_known_seeds,
                     )
 
             attn_err_handling_key_latex = (