Skip to content

Commit 48ab5b0

Browse files
brianwa84jburnim
authored andcommitted
Bump atol for vmap Beta log_prob.
PiperOrigin-RevId: 343500115
1 parent 3e2ebc4 commit 48ab5b0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tensorflow_probability/python/distributions/platform_compatibility_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@
119119
# TODO(b/142827327): Bring tolerance down to 0 for all distributions.
120120
VECTORIZED_LOGPROB_ATOL = collections.defaultdict(lambda: 1e-6)
121121
VECTORIZED_LOGPROB_ATOL.update({
122+
'Beta': 1e-5,
122123
'BetaBinomial': 1e-5,
123124
'CholeskyLKJ': 1e-4,
124125
'LKJ': 1e-3,

0 commit comments

Comments
 (0)