explicitly deal with non-finite metrics

HERA-Team · Feb 18, 2021 · 2fddde2 · 2fddde2
1 parent bc20859
commit 2fddde2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hera_qm/ant_metrics.py b/hera_qm/ant_metrics.py
@@ -494,8 +494,8 @@ def iterative_antenna_metrics_and_flagging(self, crossCut=0, deadCut=0.4, verbos
             worstCrossCutDiff = 1
 
             # Find most likely dead/crossed antenna
-            deadMetrics = self.all_metrics[iteration]['corr']
-            crossMetrics = self.all_metrics[iteration]['corrXPol']
+            deadMetrics = {ant: metric for ant, metric in self.all_metrics[iteration]['corr'].items() if np.isfinite(metric)}
+            crossMetrics = {ant: np.max(metric) for ant, metric in self.all_metrics[iteration]['corrXPol'].items() if np.isfinite(metric)}
             if (len(deadMetrics) == 0) or (len(crossMetrics) == 0):
                 break  # no unflagged antennas remain
             worstDeadAnt = min(deadMetrics, key=deadMetrics.get)