add context to transverse damper

xsuite · Jan 15, 2025 · 101345d · 101345d
1 parent 8cb5f20
commit 101345d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/xwakes/beam_elements/transverse_damper.py b/xwakes/beam_elements/transverse_damper.py
@@ -50,13 +50,16 @@ def __init__(self, gain_x, gain_y, zeta_range, num_slices,
             'py': gain_y,
         }
 
+        self.xoinitialize(**kwargs)
+
         self.slicer = xf.UniformBinSlicer(
             filling_scheme=filling_scheme,
             bunch_selection=bunch_selection,
             zeta_range=zeta_range,
             num_slices=num_slices,
             bunch_spacing_zeta=bunch_spacing_zeta,
-            moments=['px', 'py']
+            moments=['px', 'py'],
+            _context=self.context
         )
 
         if filling_scheme is not None:
@@ -74,7 +77,8 @@ def __init__(self, gain_x, gain_y, zeta_range, num_slices,
                 bunch_spacing_zeta=bunch_spacing_zeta,
                 num_periods=num_periods,
                 num_turns=1,
-                circumference=circumference
+                circumference=circumference,
+                _context=self.context
             )
 
     def _reconfigure_for_parallel(self, n_procs, my_rank):