Detect auto-correlations via np.allclose, rather than equality

ratt-ru · May 27, 2024 · 4b431c9 · 4b431c9
1 parent cbc64d7
commit 4b431c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/africanus/averaging/bda_mapping.py b/africanus/averaging/bda_mapping.py
@@ -515,7 +515,7 @@ def update_lookups(finalised, bl):
             binner.reset()
 
             # Auto-correlated baseline
-            auto_corr = ubl[bl, 0] == ubl[bl, 1]
+            auto_corr = np.allclose(ubl[bl, 0], ubl[bl, 1])
 
             for t in range(ntime):
                 # Lookup row, continue if non-existent