rough fix for very very bad chi2

Ciela-Institute · May 31, 2024 · a8471e6 · a8471e6
1 parent 30f33a1
commit a8471e6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/pqm/pqm.py b/src/pqm/pqm.py
@@ -135,7 +135,10 @@ def pqm_chi2(
     chi2_stat, _, dof, _ = _pqm_test(x_samples, y_samples, num_refs, whiten)
     if dof != num_refs - 1:
         # Rescale chi2 to new value which has the same cumulative probability
-        cp = chi2.sf(chi2_stat, dof)
-        chi2_stat = chi2.isf(cp, num_refs - 1)
+        if chi2_stat / dof < 10:
+            cp = chi2.sf(chi2_stat, dof)
+            chi2_stat = chi2.isf(cp, num_refs - 1)
+        else:
+            chi2_stat = chi2_stat * (num_refs - 1) / dof
         dof = num_refs - 1
     return chi2_stat, dof