From cd9febd7ae314c90dd11695c6733779faa4bc457 Mon Sep 17 00:00:00 2001 From: PONS Date: Wed, 22 Apr 2026 10:52:24 +0200 Subject: [PATCH] Fix private peer access using the public property --- pyaml/diagnostics/tune_monitor.py | 2 +- pyaml/tuning_tools/orbit.py | 8 ++++---- pyaml/tuning_tools/tune.py | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/pyaml/diagnostics/tune_monitor.py b/pyaml/diagnostics/tune_monitor.py index 1302ea07..1a534c1c 100644 --- a/pyaml/diagnostics/tune_monitor.py +++ b/pyaml/diagnostics/tune_monitor.py @@ -91,7 +91,7 @@ def get(self) -> np.array: rf_name = self.parent._cfg.rf_plant_name if h is not None and rf_name is not None: tune = self.parent.tune.get() - rf = self.parent._peer.get_rf_plant(rf_name) + rf = self.parent.peer.get_rf_plant(rf_name) freq = rf.frequency.get() return tune * freq / h diff --git a/pyaml/tuning_tools/orbit.py b/pyaml/tuning_tools/orbit.py index 8a4196b0..06475125 100644 --- a/pyaml/tuning_tools/orbit.py +++ b/pyaml/tuning_tools/orbit.py @@ -148,7 +148,7 @@ def correct( raise PyAMLException(f"{self.get_name()} does not have a response_matrix.") interface = pySCInterface( - element_holder=self._peer, + element_holder=self.peer, bpm_array_name=self.bpm_array_name, ) @@ -294,11 +294,11 @@ def get_rf_weight(self) -> float: return self._pySC_response_matrix.rf_weight def post_init(self): - self._hcorr = self._peer.get_magnets(self._cfg.hcorr_array_name) - self._vcorr = self._peer.get_magnets(self._cfg.vcorr_array_name) + self._hcorr = self.peer.get_magnets(self._cfg.hcorr_array_name) + self._vcorr = self.peer.get_magnets(self._cfg.vcorr_array_name) hvElts = [] hvElts.extend(self._hcorr) hvElts.extend(self._vcorr) self._hvcorr = MagnetArray("", hvElts) if self._cfg.rf_plant_name is not None: - self._rf_plant = self._peer.get_rf_plant(self._cfg.rf_plant_name) + self._rf_plant = self.peer.get_rf_plant(self._cfg.rf_plant_name) diff --git a/pyaml/tuning_tools/tune.py b/pyaml/tuning_tools/tune.py index 78e15074..a4d53b45 100644 --- a/pyaml/tuning_tools/tune.py +++ b/pyaml/tuning_tools/tune.py @@ -97,12 +97,12 @@ def load(self, load_path: Path): @property def _tm(self) -> "BetatronTuneMonitor": self.check_peer() - return self._peer.get_betatron_tune_monitor(self._cfg.betatron_tune_name) + return self.peer.get_betatron_tune_monitor(self._cfg.betatron_tune_name) @property def _quads(self) -> "MagnetArray": self.check_peer() - return self._peer.get_magnets(self._cfg.quad_array_name) + return self.peer.get_magnets(self._cfg.quad_array_name) def get(self): """