Fix bug introduced when renaming/refactoring. (#236)

ratt-ru · Mar 9, 2023 · 4e57140 · 4e57140
1 parent c3d17fc
commit 4e57140
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/quartical/gains/baseline.py b/quartical/gains/baseline.py
@@ -247,15 +247,13 @@ def apply_baseline_corrections(data_xds_list, bl_xds_list):
         bl_corrections = blxds.bl_correction.data
         ant1_col = xds.ANTENNA1.data
         ant2_col = xds.ANTENNA2.data
-        corr_mode = xds.dims["corr"]
 
         corres = da.blockwise(
             dask_apply_baseline_corrections, ("rowlike", "chan", "corr"),
             data_col, ("rowlike", "chan", "corr"),
             bl_corrections, ("rowlike", "baseline", "chan", "corr"),
             ant1_col, ("rowlike",),
             ant2_col, ("rowlike",),
-            corr_mode, None,
             dtype=data_col.dtype,
             align_arrays=False,
             concatenate=True,
@@ -278,15 +276,13 @@ def dask_apply_baseline_corrections(
     bl_corrections,
     a1,
     a2,
-    corr_mode
 ):
 
     return _apply_baseline_corrections(
         data,
         bl_corrections,
         a1,
         a2,
-        corr_mode
     )
 
 
@@ -313,7 +309,7 @@ def impl(data, bl_corrections, a1, a2):
 
                 v = data[row, f]
 
-                blg = 1/bl_corrections[bl_m, f]
+                blg = 1/bl_corrections[0, bl_m, f]
 
                 data[row, f] = blg * v