diff --git a/tensorflow_probability/examples/BUILD b/tensorflow_probability/examples/BUILD index 58c89b9220..fc7d19100a 100644 --- a/tensorflow_probability/examples/BUILD +++ b/tensorflow_probability/examples/BUILD @@ -52,6 +52,7 @@ py_library( py_test( name = "bayesian_neural_network_test", size = "medium", + timeout = "long", srcs = ["bayesian_neural_network.py"], args = [ "--fake_data",