Remove some dependencies in internal libraries on numpy backend.

PiperOrigin-RevId: 473083253

Author: srvasude

tensorflow_probability/python/internal/BUILD

@@ -344,7 +344,6 @@ multi_substrate_py_library(
     srcs = ["implementation_selection.py"],
     deps = [
         # tensorflow dep,
-        "//tensorflow_probability/python/internal/backend/numpy",
     ],
 )
 
@@ -757,7 +756,6 @@ multi_substrate_py_library(
         # tensorflow dep,
         "//tensorflow_probability/python/bijectors:bijector",
         "//tensorflow_probability/python/internal:empirical_statistical_testing",
-        "//tensorflow_probability/python/internal/backend/numpy",
         "//tensorflow_probability/python/util:deferred_tensor",
         "//tensorflow_probability/python/util:seed_stream",
     ],

tensorflow_probability/python/internal/test_util.py

@@ -34,7 +34,6 @@
 from tensorflow_probability.python.internal import empirical_statistical_testing
 from tensorflow_probability.python.internal import samplers
 from tensorflow_probability.python.internal import test_combinations
-from tensorflow_probability.python.internal.backend.numpy import ops
 from tensorflow_probability.python.util.deferred_tensor import DeferredTensor
 from tensorflow_probability.python.util.deferred_tensor import TransformedVariable
 from tensorflow_probability.python.util.seed_stream import SeedStream
@@ -868,7 +867,7 @@ def f(test_fn_or_class):
     """Decorator."""
     if JAX_MODE:
       return test_fn_or_class
-    if tf.Variable != ops.NumpyVariable:
+    if not NUMPY_MODE:
       return test_fn_or_class
 
     reason = 'Test disabled for Numpy missing functionality: {}'.format(
@@ -912,7 +911,7 @@ def tf_tape_safety_test(test_fn):
   """Only run a test of TF2 tape safety against the TF backend."""
 
   def new_test(self, *args, **kwargs):
-    if JAX_MODE or (tf.Variable == ops.NumpyVariable):
+    if JAX_MODE or NUMPY_MODE:
       self.skipTest('Tape-safety tests are only run against TensorFlow.')
     return test_fn(self, *args, **kwargs)
 

tensorflow_probability/python/math/BUILD

@@ -181,7 +181,6 @@ multi_substrate_py_library(
     deps = [
         # tensorflow dep,
         "//tensorflow_probability/python/internal:tensor_util",
-        "//tensorflow_probability/python/internal/backend/numpy:tf_inspect",
     ],
 )
 

tensorflow_probability/python/math/gradient.py

@@ -17,7 +17,7 @@
 import tensorflow.compat.v2 as tf
 
 from tensorflow_probability.python.internal import tensor_util
-from tensorflow_probability.python.internal.backend.numpy import tf_inspect
+from tensorflow.python.util import tf_inspect  # pylint: disable=g-direct-tensorflow-import
 
 
 __all__ = [