From 7314fd9bfca34f1c8d6beca07a55f856d476da16 Mon Sep 17 00:00:00 2001 From: Googler Date: Wed, 14 Aug 2024 09:45:41 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 662953761 --- tensorflow_probability/python/distributions/BUILD | 1 + 1 file changed, 1 insertion(+) 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 = [