From 526c9750aa54938076b98ab5fd3472ad45672340 Mon Sep 17 00:00:00 2001 From: Googler Date: Sun, 7 Jan 2024 15:15:22 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 596429195 --- tensorflow_probability/python/experimental/autobnn/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow_probability/python/experimental/autobnn/BUILD b/tensorflow_probability/python/experimental/autobnn/BUILD index a719518a8b..6e851b715a 100644 --- a/tensorflow_probability/python/experimental/autobnn/BUILD +++ b/tensorflow_probability/python/experimental/autobnn/BUILD @@ -175,6 +175,7 @@ py_library( py_test( name = "models_test", + timeout = "long", srcs = ["models_test.py"], shard_count = 3, deps = [