Update hypothesis test condition.

PiperOrigin-RevId: 563147229
tensorflow · Sep 6, 2023 · b15c605 · b15c605
1 parent 90a78ae
commit b15c605
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_probability/python/distributions/generalized_pareto_test.py b/tensorflow_probability/python/distributions/generalized_pareto_test.py
@@ -141,7 +141,7 @@ def testCDF(self, dist):
 
     loc, scale, conc = self.evaluate([dist.loc, dist.scale, dist.concentration])
     hp.assume(abs(loc / scale) < 1e7)
-    hp.assume(abs(conc) > 1e-12)
+    hp.assume((abs(conc) > 1e-12) or (conc == 0.))
     expected_cdf = sp_stats.genpareto(conc, loc=loc, scale=scale).cdf(xs)
     actual_cdf = self.evaluate(cdf)
     msg = ('Location: {}, scale: {}, concentration: {}, xs: {} '