From a457afab3a6407d21e83709b2e8e01066f646bb8 Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Wed, 15 Jan 2025 12:54:59 -0500
Subject: [PATCH] zdc_sigma: ignore failed fit (flaky)
#124
---
benchmarks/zdc_sigma/analysis/sigma_plots.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/benchmarks/zdc_sigma/analysis/sigma_plots.py b/benchmarks/zdc_sigma/analysis/sigma_plots.py
index 7ded93da..228dbb92 100644
--- a/benchmarks/zdc_sigma/analysis/sigma_plots.py
+++ b/benchmarks/zdc_sigma/analysis/sigma_plots.py
@@ -235,10 +235,13 @@ from scipy.optimize import curve_fit
slc=abs(bc)<0.6
fnc=gauss
p0=[100, 0, 0.5]
-coeff, var_matrix = curve_fit(fnc, bc[slc], y[slc], p0=p0,
- sigma=np.sqrt(y[slc])+(y[slc]==0), maxfev=10000)
-x=np.linspace(-1, 1)
-plt.plot(x, gauss(x, *coeff), color='tab:orange')
+try:
+ coeff, var_matrix = curve_fit(fnc, bc[slc], y[slc], p0=p0,
+ sigma=np.sqrt(y[slc])+(y[slc]==0), maxfev=10000)
+ x=np.linspace(-1, 1)
+ plt.plot(x, gauss(x, *coeff), color='tab:orange')
+except RuntimeError:
+ print("fit failed")
plt.xlabel("$\\theta^{*\\rm recon}_{\\Sigma}-\\theta^{*\\rm truth}_{\\Sigma}$ [mrad]")
plt.ylabel("events")
--
GitLab