Fix tests failing because of bug fix

desy-ml · Dec 1, 2023 · d40ec42 · d40ec42
1 parent ecc70e0
commit d40ec42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cheetah/particles.py b/cheetah/particles.py
@@ -905,7 +905,7 @@ def from_astra(cls, path: str, **kwargs) -> "ParticleBeam":
         particle_charges = torch.from_numpy(particle_charges)
         return cls(
             particles=particles_7d,
-            energy=torch.tensor(energy),
+            energy=torch.tensor(energy, dtype=torch.float32),
             particle_charges=particle_charges,
             **kwargs,
         )