Fix chi2 bug when data is missing

ptiede · Nov 8, 2024 · d5781aa · d5781aa
1 parent c75a2e2
commit d5781aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/visualizations/data.jl b/src/visualizations/data.jl
@@ -363,7 +363,10 @@ end
 
 function _chi2(res::EHTObservationTable)
     return sum(datatable(res)) do d
-            return @. abs2(d.measurement/d.noise)
+            r2 = @. abs2(d.measurement/d.noise)
+            # Check if residual is NaN which means that the data is missing
+            isnan(r2) && return zero(r2)
+            return r2
     end
 end