bugfix(prob) handle sigma > q

malb · Aug 22, 2023 · 37cd398 · 37cd398
1 parent cf36315
commit 37cd398
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)