Check if particle tracks is set

rosalindfranklininstitute · Oct 21, 2024 · 2722864 · 2722864
1 parent bbc2ee9
commit 2722864
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/parakeet/simulate/_exit_wave.py b/src/parakeet/simulate/_exit_wave.py
@@ -467,6 +467,10 @@ def get_particle_positions(particle_tracks, sample, scan):
         # Save the first position
         position0 = position.copy()
 
+        # If no tracks just return original positions
+        if particle_tracks is None:
+            return position0
+
         # For each image number and fraction update the particle position and
         # save the difference in position w.r.t the original particle position
         particle_positions = []