From e8be901da636777967626a58c1556abdda41fce3 Mon Sep 17 00:00:00 2001
From: gubaidulinvadim <gubaidulinvadim@gmail.com>
Date: Wed, 19 Jun 2024 00:01:13 +0200
Subject: [PATCH] - added a check for vertical dispersion
---
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 a5b5276..016cb9b 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:1] != 0):
+ if (self.ring.optics.local_dispersion != 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