Explicitly link modules that are imported in tests.

PiperOrigin-RevId: 476237801

Author: srvasude

tensorflow_probability/python/bijectors/BUILD

@@ -1264,6 +1264,7 @@ multi_substrate_py_test(
         ":softmax_centered",
         ":softplus",
         # absl/testing:parameterized dep,
+        # mock dep,
         # tensorflow dep,
         "//tensorflow_probability/python/internal:tensorshape_util",
         "//tensorflow_probability/python/internal:test_util",
@@ -1303,6 +1304,7 @@ multi_substrate_py_test(
         ":softmax_centered",
         ":softplus",
         ":split",
+        # mock dep,
         # numpy dep,
         # tensorflow dep,
         "//tensorflow_probability/python/internal:tensorshape_util",

tensorflow_probability/python/math/BUILD

@@ -149,7 +149,6 @@ multi_substrate_py_library(
         "//tensorflow_probability/python/internal:dtype_util",
         "//tensorflow_probability/python/internal:prefer_static",
         "//tensorflow_probability/python/internal:variadic_reduce",
-        # tensorflow/compiler/jit dep,
     ],
 )
 

tensorflow_probability/python/sts/BUILD

@@ -218,6 +218,7 @@ multi_substrate_py_test(
         ":holiday_effects",
         # holidays dep,
         # numpy dep,
+        # pandas dep,
         # tensorflow dep,
         "//tensorflow_probability/python/internal:test_util",
     ],