added tqdm

jmbhughes · Mar 8, 2021 · 5385af3 · 5385af3
1 parent a910784
commit 5385af3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/noisegate.py b/noisegate.py
@@ -1,4 +1,5 @@
 import numpy as np
+from tqdm import tqdm
 
 fft = np.fft.fftn
 ifft = np.fft.ifftn
@@ -79,7 +80,7 @@ def noise_gate(data, coords, beta_approx, gamma=3, width=12):
     gated_image = np.zeros_like(data)
 
     # over all image sections
-    for i in range(len(coords)):
+    for i in tqdm(range(len(coords))):
         x, y, t = coords[i]
 
         # get image section copy so as not to manipulate it

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 numpy
 astropy
+tqdm
-Original file line number
+Diff line change
@@ -1,2 +1,3 @@
     numpy
     astropy
+    tqdm