Unify sbend parsing fallbacks

desy-ml · Oct 7, 2024 · 5b9f2ff · 5b9f2ff
1 parent 53ecd0a
commit 5b9f2ff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cheetah/converters/elegant.py b/cheetah/converters/elegant.py
@@ -355,7 +355,7 @@ def convert_element(
                 length=torch.tensor(parsed["l"]),
                 angle=torch.tensor(parsed.get("angle", 0.0)),
                 k1=torch.tensor(parsed.get("k1", 0.0)),
-                e1=torch.tensor(parsed["e1"]),
+                e1=torch.tensor(parsed.get("e1", 0.0)),
                 e2=torch.tensor(parsed.get("e2", 0.0)),
                 tilt=torch.tensor(parsed.get("tilt", 0.0)),
                 name=name,
@@ -370,7 +370,7 @@ def convert_element(
             return cheetah.RBend(
                 length=torch.tensor(parsed["l"]),
                 angle=torch.tensor(parsed.get("angle", 0.0)),
-                e1=torch.tensor(parsed["e1"]),
+                e1=torch.tensor(parsed.get("e1", 0.0)),
                 e2=torch.tensor(parsed.get("e2", 0.0)),
                 tilt=torch.tensor(parsed.get("tilt", 0.0)),
                 name=name,
@@ -406,7 +406,7 @@ def convert_element(
                 length=torch.tensor(parsed["l"]),
                 angle=torch.tensor(parsed.get("angle", 0.0)),
                 k1=torch.tensor(parsed.get("k1", 0.0)),
-                e1=torch.tensor(parsed["e1"]),
+                e1=torch.tensor(parsed.get("e1", 0.0)),
                 e2=torch.tensor(parsed.get("e2", 0.0)),
                 tilt=torch.tensor(parsed.get("tilt", 0.0)),
                 name=name,