Mark initialization_test as size medium.

PiperOrigin-RevId: 413446007
comeetie · Dec 1, 2021 · b5f0348 · b5f0348
1 parent 224dd2c
commit b5f0348
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_probability/python/experimental/mcmc/BUILD b/tensorflow_probability/python/experimental/mcmc/BUILD
@@ -109,6 +109,7 @@ multi_substrate_py_library(
 
 multi_substrate_py_test(
     name = "initialization_test",
+    size = "medium",
     srcs = ["initialization_test.py"],
     srcs_version = "PY3",
     deps = [