diff --git a/tensorflow_probability/python/internal/backend/numpy/linalg_impl.py b/tensorflow_probability/python/internal/backend/numpy/linalg_impl.py index e69b4986ba..4080f84724 100644 --- a/tensorflow_probability/python/internal/backend/numpy/linalg_impl.py +++ b/tensorflow_probability/python/internal/backend/numpy/linalg_impl.py @@ -16,7 +16,6 @@ import collections import functools -# Dependency imports import numpy as np from tensorflow_probability.python.internal.backend.numpy import _utils as utils diff --git a/tensorflow_probability/python/internal/backend/numpy/misc.py b/tensorflow_probability/python/internal/backend/numpy/misc.py index d9f1ecdd5c..a126eb756c 100644 --- a/tensorflow_probability/python/internal/backend/numpy/misc.py +++ b/tensorflow_probability/python/internal/backend/numpy/misc.py @@ -16,7 +16,6 @@ import collections -# Dependency imports import numpy as np from tensorflow_probability.python.internal.backend.numpy import _utils as utils diff --git a/tensorflow_probability/python/internal/backend/numpy/nn.py b/tensorflow_probability/python/internal/backend/numpy/nn.py index 5609a5bcc6..3a1ce16f65 100644 --- a/tensorflow_probability/python/internal/backend/numpy/nn.py +++ b/tensorflow_probability/python/internal/backend/numpy/nn.py @@ -16,7 +16,6 @@ import collections -# Dependency imports import numpy as np from tensorflow_probability.python.internal.backend.numpy import _utils as utils diff --git a/tensorflow_probability/python/internal/backend/numpy/numpy_array.py b/tensorflow_probability/python/internal/backend/numpy/numpy_array.py index 55d46b189d..9ccb6bd995 100644 --- a/tensorflow_probability/python/internal/backend/numpy/numpy_array.py +++ b/tensorflow_probability/python/internal/backend/numpy/numpy_array.py @@ -15,7 +15,6 @@ """Numpy implementations of TensorFlow general top-level functions.""" import functools -# Dependency imports import numpy as np import numpy as onp # pylint: disable=reimported diff --git a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py index 5ef9fafd98..cae953f875 100644 --- a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py +++ b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py @@ -16,8 +16,6 @@ import functools -# Dependency imports - from absl import flags from absl import logging from absl.testing import parameterized diff --git a/tensorflow_probability/python/internal/backend/numpy/ops.py b/tensorflow_probability/python/internal/backend/numpy/ops.py index 692bd2c7c3..96e8fa5155 100644 --- a/tensorflow_probability/python/internal/backend/numpy/ops.py +++ b/tensorflow_probability/python/internal/backend/numpy/ops.py @@ -17,7 +17,6 @@ import contextlib import functools -# Dependency imports import numpy as np import numpy as onp # Avoid JAX rewrite. # pylint: disable=reimported import six diff --git a/tensorflow_probability/python/internal/backend/numpy/raw_ops.py b/tensorflow_probability/python/internal/backend/numpy/raw_ops.py index 0ccae42a60..abf5a12181 100644 --- a/tensorflow_probability/python/internal/backend/numpy/raw_ops.py +++ b/tensorflow_probability/python/internal/backend/numpy/raw_ops.py @@ -16,7 +16,6 @@ import collections -# Dependency imports import numpy as np import numpy as onp # Disable JAX rewrite. # pylint: disable=reimported diff --git a/tensorflow_probability/python/internal/backend/numpy/sets_lib.py b/tensorflow_probability/python/internal/backend/numpy/sets_lib.py index 070f685929..abd6793276 100644 --- a/tensorflow_probability/python/internal/backend/numpy/sets_lib.py +++ b/tensorflow_probability/python/internal/backend/numpy/sets_lib.py @@ -14,7 +14,6 @@ # ============================================================================ """Numpy implementations of sets functions.""" -# Dependency imports import numpy as np from tensorflow_probability.python.internal.backend.numpy import _utils as utils diff --git a/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py b/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py index 8dd6d79501..c87ae5d3d7 100644 --- a/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py +++ b/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for `internal.backend.numpy.TensorArray`.""" -# Dependency imports - import numpy as np # Rewritten by script to import jax.numpy from tensorflow_probability.python.internal.backend import numpy as nptf