desy-ml / cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
https://cheetah-accelerator.readthedocs.io
GNU General Public License v3.0
35 stars 14 forks source link

Fix corrector broadcasting bug #265

Closed jank324 closed 1 month ago

jank324 commented 1 month ago

Description

Includes .angle in the vector shape computations of VerticalCorrector and HorizontalCorrector, such that all broadcastable properties of these are now included.

Motivation and Context

Fixes #264.

Types of changes

Checklist

Note: We are using a maximum length of 88 characters per line.