bugfix(prob) fix issue #78

malb · Aug 23, 2023 · ce96585 · ce96585
1 parent cf36315
commit ce96585
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 > 16 * q:
+        return oo
+
     advantage = float(exp(-float(pi) * (float(sigma / q) ** 2)))
     return amplify(target_advantage, advantage, majority=True)