Merge pull request #40 from ChristopherMayes/fix_drift_to_zero

Fix drift to zero
ChristopherMayes · Jun 28, 2023 · 7d3a7a6 · 7d3a7a6
2 parents a1d6c9e + 2b48827
commit 7d3a7a6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 21 deletions.
diff --git a/docs/examples/particle_examples.ipynb b/docs/examples/particle_examples.ipynb
diff --git a/pmd_beamphysics/particles.py b/pmd_beamphysics/particles.py
@@ -738,7 +738,7 @@ def drift(self, delta_t):
 
     def drift_to_z(self, z=None):
 
-        if not z:
+        if z is None:
             z = self.avg('z')
         dt = (z - self.z) / (self.beta_z * c_light)
         self.drift(dt)
@@ -752,7 +752,7 @@ def drift_to_t(self, t=None):
         
         If no z is given, particles will be drifted to the average t
         """
-        if not t:
+        if t is None:
             t = self.avg('t')
         dt = t - self.t
         self.drift(dt)