Broadcasting error in corrector elements

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

34 stars 14 forks source link

Broadcasting error in corrector elements #264

Closed Hespe closed 3 weeks ago

Hespe commented 4 weeks ago

The HorizontalCorrector and VerticalCorrector do not properly broadcast their angle property if it is provided in vectorized form, leading to a crash during tracking.

Hespe commented 4 weeks ago

Relevant for #165