diff --git a/discussion/neutra/neutra_kernel_test.py b/discussion/neutra/neutra_kernel_test.py index effbf66534..debe92a71a 100644 --- a/discussion/neutra/neutra_kernel_test.py +++ b/discussion/neutra/neutra_kernel_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests NeuTra.""" -# Dependency imports from absl.testing import parameterized import tensorflow.compat.v1 as tf diff --git a/spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility_test.py b/spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility_test.py index a53f802733..f12c66b379 100644 --- a/spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility_test.py +++ b/spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility_test.py @@ -14,8 +14,6 @@ import functools -# Dependency imports - from absl import logging from absl.testing import parameterized import numpy as np diff --git a/tensorflow_probability/examples/statistical_rethinking/rethinking/quap_test.py b/tensorflow_probability/examples/statistical_rethinking/rethinking/quap_test.py index e2ee5935a1..da91f057e4 100644 --- a/tensorflow_probability/examples/statistical_rethinking/rethinking/quap_test.py +++ b/tensorflow_probability/examples/statistical_rethinking/rethinking/quap_test.py @@ -16,7 +16,6 @@ import collections -# Dependency imports from absl.testing import parameterized import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py b/tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py index fa00281b1a..fd052403e5 100644 --- a/tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py +++ b/tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py @@ -27,8 +27,6 @@ import pprint import time -# Dependency imports - import tensorflow.compat.v2 as tf RUNTIME_EAGER = 'eager' diff --git a/tensorflow_probability/python/experimental/distribute/sharded.py b/tensorflow_probability/python/experimental/distribute/sharded.py index d58e47a1a7..494b9c5eff 100644 --- a/tensorflow_probability/python/experimental/distribute/sharded.py +++ b/tensorflow_probability/python/experimental/distribute/sharded.py @@ -14,8 +14,6 @@ # ============================================================================ """Distributions for distributed computations.""" -# Dependency imports - import tensorflow.compat.v2 as tf from tensorflow_probability.python.distributions import distribution as distribution_lib diff --git a/tensorflow_probability/python/experimental/marginalize/marginalizable_test.py b/tensorflow_probability/python/experimental/marginalize/marginalizable_test.py index b0da46d476..e3ee64ef01 100644 --- a/tensorflow_probability/python/experimental/marginalize/marginalizable_test.py +++ b/tensorflow_probability/python/experimental/marginalize/marginalizable_test.py @@ -21,8 +21,6 @@ import functools import itertools -# Dependency imports - import numpy as np import tensorflow.compat.v2 as tf from tensorflow_probability.python.distributions import bernoulli diff --git a/tensorflow_probability/python/experimental/mcmc/internal/test_fixtures.py b/tensorflow_probability/python/experimental/mcmc/internal/test_fixtures.py index af633f9561..41de535eb6 100644 --- a/tensorflow_probability/python/experimental/mcmc/internal/test_fixtures.py +++ b/tensorflow_probability/python/experimental/mcmc/internal/test_fixtures.py @@ -16,7 +16,6 @@ import collections -# Dependency imports import tensorflow.compat.v2 as tf from tensorflow_probability.python.experimental.mcmc import reducer from tensorflow_probability.python.mcmc import kernel diff --git a/tensorflow_probability/python/experimental/nn/util/convolution_util_test.py b/tensorflow_probability/python/experimental/nn/util/convolution_util_test.py index d86b31a3cd..d648b27496 100644 --- a/tensorflow_probability/python/experimental/nn/util/convolution_util_test.py +++ b/tensorflow_probability/python/experimental/nn/util/convolution_util_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for batched convolutions.""" -# Dependency imports - from absl.testing import parameterized import numpy as np diff --git a/tensorflow_probability/python/experimental/nn/util/kernel_bias_test.py b/tensorflow_probability/python/experimental/nn/util/kernel_bias_test.py index d571a4cc86..82241f1955 100644 --- a/tensorflow_probability/python/experimental/nn/util/kernel_bias_test.py +++ b/tensorflow_probability/python/experimental/nn/util/kernel_bias_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for kernel_bias.""" -# Dependency imports - from tensorflow_probability.python.experimental.nn.util import kernel_bias from tensorflow_probability.python.internal import test_util diff --git a/tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter_test.py b/tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter_test.py index ba21da8732..28f058095a 100644 --- a/tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter_test.py +++ b/tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter_test.py @@ -15,8 +15,6 @@ """Tests for the Ensemble Adjustment Kalman Filter.""" -# Dependency imports - import tensorflow.compat.v2 as tf from tensorflow_probability.python.distributions import joint_distribution_named as jdn diff --git a/tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter_test.py b/tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter_test.py index 1f2911217c..eee8ae0078 100644 --- a/tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter_test.py +++ b/tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter_test.py @@ -16,7 +16,6 @@ import collections -# Dependency imports from absl.testing import parameterized import numpy as np diff --git a/tensorflow_probability/python/experimental/util/trainable.py b/tensorflow_probability/python/experimental/util/trainable.py index 6dea8680db..5172f701ac 100644 --- a/tensorflow_probability/python/experimental/util/trainable.py +++ b/tensorflow_probability/python/experimental/util/trainable.py @@ -16,7 +16,6 @@ import functools -# Dependency imports import tensorflow.compat.v2 as tf from tensorflow_probability.python.internal import docstring_util diff --git a/tensorflow_probability/python/internal/backend/jax/rewrite.py b/tensorflow_probability/python/internal/backend/jax/rewrite.py index 68efbd20f5..a3b0703672 100644 --- a/tensorflow_probability/python/internal/backend/jax/rewrite.py +++ b/tensorflow_probability/python/internal/backend/jax/rewrite.py @@ -14,8 +14,6 @@ # ============================================================================ """Rewrite script for NP->JAX.""" -# Dependency imports - from absl import app from absl import flags diff --git a/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py b/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py index 68932e5f95..8974b8b3af 100644 --- a/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py +++ b/tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py @@ -15,8 +15,6 @@ import gc import operator -# Dependency imports - from absl.testing import parameterized import numpy as np import tensorflow.compat.v1 as tf1 diff --git a/tensorflow_probability/python/math/ode/xla_test.py b/tensorflow_probability/python/math/ode/xla_test.py index 44987bcf1e..f06cd894d2 100644 --- a/tensorflow_probability/python/math/ode/xla_test.py +++ b/tensorflow_probability/python/math/ode/xla_test.py @@ -14,7 +14,6 @@ # ============================================================================ """XLA tests for TensorFlow Probability ODE solvers.""" -# Dependency imports from absl import flags from absl.testing import parameterized import numpy as np diff --git a/tensorflow_probability/python/math/psd_kernels/internal/util_test.py b/tensorflow_probability/python/math/psd_kernels/internal/util_test.py index 7ce4de5dab..6a63c911c0 100644 --- a/tensorflow_probability/python/math/psd_kernels/internal/util_test.py +++ b/tensorflow_probability/python/math/psd_kernels/internal/util_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for Positive-Semidefinite Kernels utilities.""" -# Dependency imports from absl.testing import parameterized import numpy as np import tensorflow.compat.v1 as tf1 diff --git a/tensorflow_probability/python/monte_carlo/expectation_test.py b/tensorflow_probability/python/monte_carlo/expectation_test.py index 3e6d2c6e0c..6504af1a9a 100644 --- a/tensorflow_probability/python/monte_carlo/expectation_test.py +++ b/tensorflow_probability/python/monte_carlo/expectation_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for Monte Carlo Ops.""" -# Dependency imports - import tensorflow.compat.v2 as tf from tensorflow_probability.python.distributions import bernoulli from tensorflow_probability.python.distributions import joint_distribution_named as jdn diff --git a/tensorflow_probability/python/random/random_ops_test.py b/tensorflow_probability/python/random/random_ops_test.py index 4453dea2d6..9dbe8a6c6c 100644 --- a/tensorflow_probability/python/random/random_ops_test.py +++ b/tensorflow_probability/python/random/random_ops_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for generating random samples.""" -# Dependency imports import numpy as np import tensorflow.compat.v1 as tf1 diff --git a/tensorflow_probability/substrates/meta/rewrite.py b/tensorflow_probability/substrates/meta/rewrite.py index f856bdaf1a..c2392254d3 100644 --- a/tensorflow_probability/substrates/meta/rewrite.py +++ b/tensorflow_probability/substrates/meta/rewrite.py @@ -16,7 +16,6 @@ import collections -# Dependency imports from absl import app from absl import flags