insert_neutron: ignore failed fit (flaky) (#124)

eic · Jan 10, 2025 · d87ae97 · d87ae97
1 parent 6de018e
commit d87ae97
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/benchmarks/insert_neutron/analysis/neutron_plots.py b/benchmarks/insert_neutron/analysis/neutron_plots.py
@@ -79,11 +79,12 @@ def gauss(x, A,mu, sigma):
 fnc=gauss
 sigma=np.sqrt(y[slc])+(y[slc]==0)
 p0=(100, 0, 5)
-coeff, var_matrix = curve_fit(fnc, list(bc[slc]), list(y[slc]), p0=p0, sigma=list(sigma), maxfev=10000)
-xx=np.linspace(-5,5,100)
-plt.plot(xx,fnc(xx,*coeff))
-# except:
-#     pass
+try:
+    coeff, var_matrix = curve_fit(fnc, list(bc[slc]), list(y[slc]), p0=p0, sigma=list(sigma), maxfev=10000)
+    xx=np.linspace(-5,5,100)
+    plt.plot(xx,fnc(xx,*coeff))
+except RuntimeError:
+    print("fit failed")
 plt.xlabel("$\\theta_{rec}-\\theta_{truth}$ [mrad]")
 plt.ylabel("events")
 plt.title(f"$p={p}$ GeV, ${eta_min}<\\eta<{eta_max}$")