Update visibility.py

Change `use_tqdm` variable name to `show_progress`
radionets-project · Jul 18, 2024 · 79ddd58 · 79ddd58
1 parent 7a35727
commit 79ddd58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyvisgen/simulation/visibility.py b/pyvisgen/simulation/visibility.py
@@ -39,7 +39,7 @@ def add(self, visibilities):
         ]
 
 
-def vis_loop(obs, SI, num_threads=10, noisy=True, mode="full", batch_size=100, use_tqdm=False):
+def vis_loop(obs, SI, num_threads=10, noisy=True, mode="full", batch_size=100, show_progress=False):
     torch.set_num_threads(num_threads)
     torch._dynamo.config.suppress_errors = True
 
@@ -97,7 +97,7 @@ def vis_loop(obs, SI, num_threads=10, noisy=True, mode="full", batch_size=100, u
 
     batches = torch.arange(bas[:].shape[1]).split(batch_size)
 
-    if use_tqdm:
+    if show_progress:
         batches = tqdm(batches)
 
     for p in batches:
@@ -170,4 +170,4 @@ def generate_noise(shape, obs, SEFD):
     noise = torch.normal(mean=0, std=std, size=shape, device=obs.device)
     noise = noise + 1.0j * torch.normal(mean=0, std=std, size=shape, device=obs.device)
 
-    return noise
+    return noise