Explicitly computing array indices for baselines in fringefit code

Author: dessmalljive

docs/theory/fringefit-snr.ipynb

@@ -890,7 +890,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.13.7"
   }
  },
  "nbformat": 4,

src/astroviper/calibration/fringefit.py

@@ -65,8 +65,8 @@ def _fringe_node_task(input_params: Dict):
     xds2 = xds.isel(**data_selection[name])
     ddelay, drate = getFourierSpacings(xds2, npad)
     baselines = xds2.baseline_id[
-        (xds2.baseline_id.baseline_antenna1_name == ref_ant)
-        | (xds2.baseline_id.baseline_antenna2_name == ref_ant)
+        (xds2.baseline_id.baseline_antenna1_name == ref_ant).compute()
+        | (xds2.baseline_id.baseline_antenna2_name == ref_ant).compute()
     ]
     vis = xds2.VISIBILITY[:, baselines, :, ::3]
     flags = xds2.FLAG[:, baselines, :, ::3]

tests/calibration/test_fringefit_sbd_fft.py

@@ -43,9 +43,9 @@
 
 
 def bl_inds(xds, a1, a2):
-    return (xds.baseline_antenna1_name == a1) & (xds.baseline_antenna2_name == a2) | (
-        xds.baseline_antenna1_name == a2
-    ) & (xds.baseline_antenna2_name == a1)
+    return (
+        (xds.baseline_antenna1_name == a1) & (xds.baseline_antenna2_name == a2) |
+        (xds.baseline_antenna1_name == a2) & (xds.baseline_antenna2_name == a1)).compute()
 
 
 # We need an xds to get baselines from; I happen to know this one contains the interval we chose above