Add missing argument entry in docstring

desy-ml · Jan 17, 2025 · 3ace4b3 · 3ace4b3
1 parent 15cc15c
commit 3ace4b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cheetah/accelerator/element.py b/cheetah/accelerator/element.py
@@ -51,6 +51,7 @@ def transfer_map(
         affine transformation.
 
         :param energy: Reference energy of the Beam. Read from the fed-in Cheetah Beam.
+        :param particle_mass_eV: Mass of the beam's particle species in eV.
         :return: A 7x7 Matrix for further calculations.
         """
         raise NotImplementedError