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,