From 17cc66c2c410b24721bff9af4b2a3d29dcd7916c Mon Sep 17 00:00:00 2001
From: colcarroll <colcarroll@google.com>
Date: Wed, 17 Jan 2024 12:38:22 -0800
Subject: [PATCH] Disable autobnn tests on kokoro.

PiperOrigin-RevId: 599256837
---
 tensorflow_probability/python/experimental/autobnn/BUILD | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow_probability/python/experimental/autobnn/BUILD b/tensorflow_probability/python/experimental/autobnn/BUILD
index 6e851b715a..db2c285237 100644
--- a/tensorflow_probability/python/experimental/autobnn/BUILD
+++ b/tensorflow_probability/python/experimental/autobnn/BUILD
@@ -39,6 +39,8 @@ py_library(
 py_test(
     name = "bnn_test",
     srcs = ["bnn_test.py"],
+    # TODO(b/320723712): enable this test in OSS.
+    tags = ["no-oss-ci"],
     deps = [
         ":bnn",
         # absl/testing:absltest dep,