Revert "Don't imply that you shouldn't use set_quadrupoles"

This reverts commit b7ed85e.
RL4AA · Nov 19, 2023 · a4c02a4 · a4c02a4
1 parent b7ed85e
commit a4c02a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/environments/awake_e_steering.py b/src/environments/awake_e_steering.py
@@ -294,7 +294,7 @@ def get_bpms(self) -> np.ndarray:
             ]
         )
 
-    def set_quadrupoles(self, values: np.ndarray) -> None:
+    def _set_quadrupoles(self, values: np.ndarray) -> None:
         for quad, setting in zip(self._quads, values):
             quad.k1 = torch.as_tensor(setting.astype(np.float32))
 
@@ -329,7 +329,7 @@ def update(self) -> None:
         drifted_and_disturbed_quad_settings = (
             drifted_quad_settings * self._random_quad_disturbance_factors
         )
-        self.set_quadrupoles(drifted_and_disturbed_quad_settings)
+        self._set_quadrupoles(drifted_and_disturbed_quad_settings)
 
         _ = self.segment.track(self.incoming)