diff --git a/tensorflow_probability/python/layers/conv_variational_test.py b/tensorflow_probability/python/layers/conv_variational_test.py index 3822257aa2..27dbfb9712 100644 --- a/tensorflow_probability/python/layers/conv_variational_test.py +++ b/tensorflow_probability/python/layers/conv_variational_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for convolutional variational layers.""" # pylint: disable=g-import-not-at-top -# Dependency imports try: from keras.testing_infra import test_utils as keras_test_utils except ImportError: diff --git a/tensorflow_probability/python/layers/dense_variational_test.py b/tensorflow_probability/python/layers/dense_variational_test.py index 7f06b1ade5..c137605319 100644 --- a/tensorflow_probability/python/layers/dense_variational_test.py +++ b/tensorflow_probability/python/layers/dense_variational_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for dense variational layers.""" # pylint: disable=g-import-not-at-top -# Dependency imports try: from keras.testing_infra import test_utils as keras_test_utils except ImportError: diff --git a/tensorflow_probability/python/layers/dense_variational_v2_test.py b/tensorflow_probability/python/layers/dense_variational_v2_test.py index 51c61d9fae..9a54d3e0c9 100644 --- a/tensorflow_probability/python/layers/dense_variational_v2_test.py +++ b/tensorflow_probability/python/layers/dense_variational_v2_test.py @@ -14,8 +14,6 @@ # ============================================================================ """Tests for tensorflow_probability.layers.DenseVariational.""" -# Dependency imports - import numpy as np import tensorflow.compat.v2 as tf from tensorflow_probability.python.distributions import distribution diff --git a/tensorflow_probability/python/layers/distribution_layer.py b/tensorflow_probability/python/layers/distribution_layer.py index 638d15e61d..f62dfed642 100644 --- a/tensorflow_probability/python/layers/distribution_layer.py +++ b/tensorflow_probability/python/layers/distribution_layer.py @@ -20,7 +20,6 @@ import io import pickle -# Dependency imports from cloudpickle import CloudPickler import numpy as np import six diff --git a/tensorflow_probability/python/layers/distribution_layer_test.py b/tensorflow_probability/python/layers/distribution_layer_test.py index 197f889a7f..bcdbef21f9 100644 --- a/tensorflow_probability/python/layers/distribution_layer_test.py +++ b/tensorflow_probability/python/layers/distribution_layer_test.py @@ -14,8 +14,6 @@ # ============================================================================ import functools -# Dependency imports - import numpy as np import six import tensorflow.compat.v1 as tf1 diff --git a/tensorflow_probability/python/layers/initializers.py b/tensorflow_probability/python/layers/initializers.py index c0b57bfddb..19cb1db11f 100644 --- a/tensorflow_probability/python/layers/initializers.py +++ b/tensorflow_probability/python/layers/initializers.py @@ -14,7 +14,6 @@ # ============================================================================ """Keras initializers useful for TFP Keras layers.""" -# Dependency imports import numpy as np import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/layers/initializers_test.py b/tensorflow_probability/python/layers/initializers_test.py index dc451cee26..437b5062a8 100644 --- a/tensorflow_probability/python/layers/initializers_test.py +++ b/tensorflow_probability/python/layers/initializers_test.py @@ -14,7 +14,6 @@ # ============================================================================ """Tests for tensorflow_probability.layers Keras initializers.""" -# Dependency imports import numpy as np from tensorflow_probability.python.internal import test_util diff --git a/tensorflow_probability/python/layers/masked_autoregressive.py b/tensorflow_probability/python/layers/masked_autoregressive.py index 07a406ec5a..e24c6ec667 100644 --- a/tensorflow_probability/python/layers/masked_autoregressive.py +++ b/tensorflow_probability/python/layers/masked_autoregressive.py @@ -14,7 +14,6 @@ # ============================================================================ """Layers for normalizing flows and masked autoregressive density estimation.""" -# Dependency imports import tensorflow.compat.v2 as tf from tensorflow_probability.python.bijectors import masked_autoregressive as masked_autoregressive_lib diff --git a/tensorflow_probability/python/layers/masked_autoregressive_test.py b/tensorflow_probability/python/layers/masked_autoregressive_test.py index 24b382ffba..951ac367c7 100644 --- a/tensorflow_probability/python/layers/masked_autoregressive_test.py +++ b/tensorflow_probability/python/layers/masked_autoregressive_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================ -# Dependency imports - import numpy as np import tensorflow.compat.v2 as tf from tensorflow_probability.python.bijectors import masked_autoregressive as masked_autoregressive_lib diff --git a/tensorflow_probability/python/layers/util.py b/tensorflow_probability/python/layers/util.py index c8b607f3c1..db4a27051b 100644 --- a/tensorflow_probability/python/layers/util.py +++ b/tensorflow_probability/python/layers/util.py @@ -19,7 +19,6 @@ import marshal import os import types -# Dependency imports import numpy as np import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/layers/variable_input.py b/tensorflow_probability/python/layers/variable_input.py index 0dae6ff7ef..6e7bf79726 100644 --- a/tensorflow_probability/python/layers/variable_input.py +++ b/tensorflow_probability/python/layers/variable_input.py @@ -14,7 +14,6 @@ # ============================================================================ """VariableInputLayer.""" -# Dependency imports import numpy as np import tensorflow.compat.v2 as tf diff --git a/tensorflow_probability/python/layers/weight_norm_test.py b/tensorflow_probability/python/layers/weight_norm_test.py index bdbda2fa49..e4b69a54ee 100644 --- a/tensorflow_probability/python/layers/weight_norm_test.py +++ b/tensorflow_probability/python/layers/weight_norm_test.py @@ -16,8 +16,6 @@ import tempfile -# Dependency imports - from absl.testing import parameterized import numpy as np