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

35 stars 14 forks source link

Broadcasting error in corrector elements #264

Closed Hespe closed 1 month ago

Hespe commented 1 month 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 1 month ago

Relevant for #165