Avoid overflow errors

JasonGross · JasonGross · commit fdc7584dcc7e · 2024-12-09T15:12:52.000-08:00
diff --git a/notebooks_jason/max_of_K_all_models.py b/notebooks_jason/max_of_K_all_models.py
@@ -3659,7 +3659,10 @@ def do_linear_regression(X, Y):
                     )
                     if include_baseline:
                         x_vals = np.logspace(
-                            np.log2(min_flop), np.log2(max_flop), 100, base=2
+                            np.log2(float(min_flop)),
+                            np.log2(float(max_flop)),
+                            100,
+                            base=2,
                         )
                         y_vals = brute_force_slope * x_vals
                         match PLOT_WITH:

Original file line number	Diff line number	Diff line change
`@@ -3659,7 +3659,10 @@ def do_linear_regression(X, Y):`
`3659`	`3659`	`)`
`3660`	`3660`	`if include_baseline:`
`3661`	`3661`	`x_vals = np.logspace(`
`3662`		`- np.log2(min_flop), np.log2(max_flop), 100, base=2`
	`3662`	`+ np.log2(float(min_flop)),`
	`3663`	`+ np.log2(float(max_flop)),`
	`3664`	`+ 100,`
	`3665`	`+ base=2,`
`3663`	`3666`	`)`
`3664`	`3667`	`y_vals = brute_force_slope * x_vals`
`3665`	`3668`	`match PLOT_WITH:`