Merge pull request #43 from radionets-project/normalization

Add normalization bool parameter to visibility calculations

Author: tgross03

docs/changes/43.optimization.rst

@@ -0,0 +1 @@
+Added optional `normalize` parameter to `pyvisgen.visibility.vis_loop` to decide whether to apply a normalization multiplier of `0.5` (default: True)

pyvisgen/simulation/visibility.py

@@ -47,6 +47,7 @@ def vis_loop(
     mode="full",
     batch_size="auto",
     show_progress=False,
+    normalize=True,
 ):
     torch.set_num_threads(num_threads)
     torch._dynamo.config.suppress_errors = True
@@ -81,7 +82,8 @@ def vis_loop(
 
     # normalize visibilities to factor 0.5,
     # so that the Stokes I image is normalized to 1
-    B *= 0.5
+    if normalize:
+        B *= 0.5
 
     # calculate vis
     visibilities = Visibilities(