bugfix(prob) WIP: handle sigma > q -- need a better condition

malb · Aug 22, 2023 · 44f0716 · 44f0716
1 parent cf36315
commit 44f0716
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/estimator/prob.py b/estimator/prob.py
@@ -124,5 +124,8 @@ def amplify_sigma(target_advantage, sigma, q):
         sigma = sum(sigma_**2 for sigma_ in sigma).sqrt()
     except TypeError:
         pass
+    if sigma > q:
+        return oo
+
     advantage = float(exp(-float(pi) * (float(sigma / q) ** 2)))
     return amplify(target_advantage, advantage, majority=True)