Complete defining features

desy-ml · Aug 26, 2024 · f766cf5 · f766cf5
1 parent 46f9fe9
commit f766cf5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cheetah/accelerator/dipole.py b/cheetah/accelerator/dipole.py
@@ -444,14 +444,20 @@ def __repr__(self):
     def defining_features(self) -> list[str]:
         return super().defining_features + [
             "length",
+            "p0c",
             "angle",
             "k1",
             "e1",
             "e2",
+            "tilt",
+            "gap",
+            "gap_exit",
             "fringe_integral",
             "fringe_integral_exit",
-            "gap",
-            "tilt",
+            "fringe_at",
+            "fringe_type",
+            "tracking_method",
+            "name",
         ]
 
     def plot(self, ax: plt.Axes, s: float) -> None: