Making changes in response to comments

RadioAstronomySoftwareGroup · Jun 28, 2024 · 6bd6834 · 6bd6834
1 parent 177fbe1
commit 6bd6834
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/pyuvdata/uvdata/mir_parser.py b/src/pyuvdata/uvdata/mir_parser.py
@@ -2100,9 +2100,9 @@ def _fix_acdata(self):
             data_arr = self.ac_data._data[field]
             export_data_arr = self.bl_data._data[field]
             for key, idx_arr in ac_groups.items():
-                try:
+                if key in bl_groups:
                     data_arr[idx_arr] = np.median(export_data_arr[bl_groups[key]])
-                except KeyError:
+                else:
                     # If no key is found, then skip over filling in the data. Odds are
                     # this is single-rx data, which means the second receiver is
                     # basically blank, so flag the entry

diff --git a/tests/uvdata/test_mir_parser.py b/tests/uvdata/test_mir_parser.py
@@ -2364,4 +2364,3 @@ def test_mir_parser_missing_auto_rx(mir_data):
 
     # Verify that every other entry is flagged, which means no data for that rx
     assert np.array_equal(mir_data.ac_data._mask, [True, False] * 8)
-    mir_data.ac_data["corrchunk"]