From 5f9a437d0bb1c79f2840b30839c08fd8ca4a3c62 Mon Sep 17 00:00:00 2001 From: Jan Kaiser Date: Tue, 1 Jul 2025 18:03:09 +0200 Subject: [PATCH] Clone everything for `Drift.track` in `"cheetah"` mode --- cheetah/accelerator/element.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cheetah/accelerator/element.py b/cheetah/accelerator/element.py index 9488a78a6..8f2134d26 100644 --- a/cheetah/accelerator/element.py +++ b/cheetah/accelerator/element.py @@ -92,8 +92,8 @@ def track(self, incoming: Beam) -> Beam: return ParameterBeam( new_mu, new_cov, - incoming.energy, - total_charge=incoming.total_charge, + incoming.energy.clone(), + total_charge=incoming.total_charge.clone(), s=new_s, species=incoming.species.clone(), ) @@ -103,9 +103,9 @@ def track(self, incoming: Beam) -> Beam: new_s = incoming.s + self.length return ParticleBeam( new_particles, - incoming.energy, - particle_charges=incoming.particle_charges, - survival_probabilities=incoming.survival_probabilities, + incoming.energy.clone(), + particle_charges=incoming.particle_charges.clone(), + survival_probabilities=incoming.survival_probabilities.clone(), s=new_s, species=incoming.species.clone(), )