From e1a7d5479966ab3ef0becfdb4b190a1dc1877eb1 Mon Sep 17 00:00:00 2001
From: gubaidulinvadim <gubaidulinvadim@gmail.com>
Date: Mon, 15 Jul 2024 15:57:58 +0200
Subject: [PATCH] forced local_dispersion to be a numpy array for comparisons
---
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 e3c6f2e..190a397 100644
--- a/mbtrack2/tracking/particles.py
+++ b/mbtrack2/tracking/particles.py
@@ -287,7 +287,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, 0, 0, 0]):
+ if (np.array(self.ring.optics.local_dispersion) != np.array([0, 0, 0, 0])).all():
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