diff --git a/tensorflow_probability/python/distributions/generalized_pareto_test.py b/tensorflow_probability/python/distributions/generalized_pareto_test.py index 16a1d2209a..0c30b71124 100644 --- 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: {} '