From 666bacbb08a12cfba46ea74be02ded86875cc3f0 Mon Sep 17 00:00:00 2001
From: GUBAIDULIN <vadim.gubaidulin@synchrotron-soleil.fr>
Date: Wed, 19 Jun 2024 14:43:50 +0200
Subject: [PATCH] Agreed
---
mbtrack2/tracking/particles.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mbtrack2/tracking/particles.py b/mbtrack2/tracking/particles.py
index 016cb9b..7c42a5b 100644
--- a/mbtrack2/tracking/particles.py
+++ b/mbtrack2/tracking/particles.py
@@ -288,7 +288,7 @@ class Bunch:
cov_y = np.cov(self['y'], self['yp'])
cov_z = np.cov(self['tau'], self['delta'])
- if (self.ring.optics.local_dispersion != 0):
+ if (self.ring.optics.local_dispersion != 0).any():
cov_xdelta = np.cov(self['x'], self['delta'])
cov_xpdelta = np.cov(self['xp'], self['delta'])
cov_ydelta = np.cov(self['y'], self['delta'])
--
GitLab