From 3bd671de54d9b167df66bba998ca0a58c35fe141 Mon Sep 17 00:00:00 2001 From: Jacob Burnim Date: Mon, 11 Mar 2024 05:25:37 +0000 Subject: [PATCH] Skip OSS tests for two benchmarks with no test cases. --- tensorflow_probability/python/mcmc/BUILD | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow_probability/python/mcmc/BUILD b/tensorflow_probability/python/mcmc/BUILD index 0e7b126578..7afcb9bca5 100644 --- a/tensorflow_probability/python/mcmc/BUILD +++ b/tensorflow_probability/python/mcmc/BUILD @@ -615,6 +615,7 @@ multi_substrate_py_test( py_library( name = "eight_schools_hmc", srcs = ["eight_schools_hmc.py"], + tags = ["no-oss-ci"], deps = [ ":hmc", ":sample", @@ -630,6 +631,7 @@ py_test( name = "eight_schools_hmc_eager_test", size = "medium", # Might run > 1 minute. srcs = ["eight_schools_hmc_eager_test.py"], + tags = ["no-oss-ci"], deps = [ ":eight_schools_hmc", # tensorflow dep,