diff --git a/tensorflow_probability/python/bijectors/BUILD b/tensorflow_probability/python/bijectors/BUILD index 14b1dc619f..55c15528fa 100644 --- a/tensorflow_probability/python/bijectors/BUILD +++ b/tensorflow_probability/python/bijectors/BUILD @@ -1854,6 +1854,7 @@ multi_substrate_py_test( multi_substrate_py_test( name = "real_nvp_test", size = "small", + timeout = "moderate", srcs = ["real_nvp_test.py"], deps = [ ":inline",