diff --git a/benchmarks/insert_tau/analysis/tau_plots.py b/benchmarks/insert_tau/analysis/tau_plots.py index f9f5c86..77c2dd3 100644 --- a/benchmarks/insert_tau/analysis/tau_plots.py +++ b/benchmarks/insert_tau/analysis/tau_plots.py @@ -135,7 +135,7 @@ def gauss(x, A,mu, sigma): fnc=gauss sigma=np.sqrt(y[slc])+(y[slc]==0) - coeff, var_matrix = curve_fit(fnc, list(bc[slc]), list(y[slc]), p0=p0,sigma=list(sigma)) + coeff, var_matrix = curve_fit(fnc, list(bc[slc]), list(y[slc]), p0=p0, sigma=list(sigma), maxfev=10000) res.append(abs(coeff[2])) dres.append(np.sqrt(var_matrix[2][2])) emid.append((emin+emax)/2) @@ -147,7 +147,7 @@ def gauss(x, A,mu, sigma): fnc=lambda E,B:B/E p0=[1,] -coeff, var_matrix = curve_fit(fnc, emid, res, p0=p0,sigma=list(dres)) +coeff, var_matrix = curve_fit(fnc, emid, res, p0=p0, sigma=list(dres), maxfev=10000) xx=np.linspace(10, 100, 100) plt.plot(xx, 1000*fnc(xx, *coeff), label=f"fit: ${coeff[0]:.2f}/E$ mrad") plt.legend()