diff --git a/tensorflow_probability/python/distributions/BUILD b/tensorflow_probability/python/distributions/BUILD index 413f625142..ee9f605d78 100644 --- a/tensorflow_probability/python/distributions/BUILD +++ b/tensorflow_probability/python/distributions/BUILD @@ -2658,6 +2658,7 @@ multi_substrate_py_test( multi_substrate_py_test( name = "beta_test", + timeout = "moderate", srcs = ["beta_test.py"], jax_size = "medium", deps = [