Fix issue with lengthless fix for Screen

desy-ml · Apr 6, 2024 · 806d191 · 806d191
1 parent 6faef7d
commit 806d191
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cheetah/accelerator.py b/cheetah/accelerator.py
@@ -1393,6 +1393,7 @@ def __init__(
             if misalignment is not None
             else torch.tensor([(0.0, 0.0)], **factory_kwargs)
         )
+        self.length = torch.zeros(self.misalignment.shape[:-1], **factory_kwargs)
         self.is_active = is_active
 
         self.set_read_beam(None)