Fix key error bug

JasonGross · JasonGross · commit 4ab2a73088cd · 2024-11-06T13:07:04.000-08:00
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 = (

Original file line number	Diff line number	Diff line change
`@@ -3266,6 +3266,7 @@ def do_linear_regression(X, Y):`
`3266`	`3266`	`]`
`3267`	`3267`	`print(f"len after: {len(df)}")`
`3268`	`3268`
	`3269`	`+ sorted_known_seeds = tuple(sorted(df["seed"].values))`
`3269`	`3270`	`# Group by 'attention_error_handling' and calculate the max 'normalized-accuracy-bound' for sorting groups`
`3270`	`3271`	`df = df[`
`3271`	`3272`	`[`
`@@ -3329,7 +3330,7 @@ def do_linear_regression(X, Y):`
`3329`	`3330`	`sing_upper_bound,`
`3330`	`3331`	`attn_err_handling_key,`
`3331`	`3332`	`best_bound_only,`
`3332`		`- tuple(sorted(subgroup["seed"].values)),`
	`3333`	`+ sorted_known_seeds,`
`3333`	`3334`	`)`
`3334`	`3335`
`3335`	`3336`	`attn_err_handling_key_latex = (`