Fix issue with lengthless fix for Screen

jank324 · jank324 · commit 806d191ed012 · 2024-04-06T14:15:28.000+02:00
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)

Original file line number	Diff line number	Diff line change
`@@ -1393,6 +1393,7 @@ def __init__(`
`1393`	`1393`	`if misalignment is not None`
`1394`	`1394`	`else torch.tensor([(0.0, 0.0)], **factory_kwargs)`
`1395`	`1395`	`)`
	`1396`	`+ self.length = torch.zeros(self.misalignment.shape[:-1], **factory_kwargs)`
`1396`	`1397`	`self.is_active = is_active`
`1397`	`1398`
`1398`	`1399`	`self.set_read_beam(None)`