From 4f07219d4372debb064c2188e92458b3129881b2 Mon Sep 17 00:00:00 2001
From: gubaidulinvadim <gubaidulinvadim@gmail.com>
Date: Mon, 15 Jul 2024 22:22:31 +0200
Subject: [PATCH] np.any() to trigger emittance calculation with dispersion
instead of np.all()
---
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 0d819a8..2b3edbb 100644
--- a/mbtrack2/tracking/particles.py
+++ b/mbtrack2/tracking/particles.py
@@ -288,7 +288,7 @@ class Bunch:
cov_z = np.cov(self['tau'], self['delta'])
if (np.array(self.ring.optics.local_dispersion)
- != np.array([0, 0, 0, 0])).all():
+ != np.array([0, 0, 0, 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