Strip tensorflow_no_contrib from Jax substrate deps.

PiperOrigin-RevId: 814328384

Author: AdamHillier

tensorflow_probability/python/build_defs.bzl

@@ -216,6 +216,7 @@ def multi_substrate_py_library(
     remove_deps = [
         "//third_party/py/tensorflow",
         "//third_party/py/tensorflow:tensorflow",
+        "//third_party/py/tensorflow:tensorflow_no_contrib",
         "//tensorflow_probability/python/internal:tf_keras",
     ]
 
@@ -342,6 +343,7 @@ def multi_substrate_py_test(
     remove_deps = [
         "//third_party/py/tensorflow",
         "//third_party/py/tensorflow:tensorflow",
+        "//third_party/py/tensorflow:tensorflow_no_contrib",
         "//tensorflow_probability/python/internal:tf_keras",
     ]
 

tensorflow_probability/substrates/meta/BUILD

@@ -32,6 +32,7 @@ py_test(
     srcs = ["numpy_integration_test.py"],
     tags = ["tfp_numpy"],
     deps = [
+        # absl/testing:absltest dep,
         "//tensorflow_probability/substrates:numpy",
     ],
 )
@@ -41,6 +42,7 @@ py_test(
     srcs = ["jax_integration_test.py"],
     tags = ["tfp_jax"],
     deps = [
+        # absl/testing:absltest dep,
         # jax dep,
         "//tensorflow_probability/substrates:jax",
     ],