From bb0b9b21a44b5a2ff559dc2313add282a75aaaef Mon Sep 17 00:00:00 2001
From: gubaidulinvadim <gubaidulinvadim@gmail.com>
Date: Mon, 15 Jul 2024 14:49:04 +0200
Subject: [PATCH] bug: corrected check if dispersion is 0 or not
---
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 8a50004..db333b3 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).any():
+ if (self.ring.optics.local_dispersion != [0, 0, 0, 0]):
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