Reformatting the code base with isort.

PiperOrigin-RevId: 451564271

keras/__init__.py

@@ -17,16 +17,17 @@
 Detailed documentation and user guides are available at
 [keras.io](https://keras.io).
 """
-# pylint: disable=unused-import
-from tensorflow.python import tf2
-from tensorflow.python.util.tf_export import keras_export
-
 from keras import distribute
 from keras import models
 from keras.engine.input_layer import Input
 from keras.engine.sequential import Sequential
 from keras.engine.training import Model
 
+# isort: off
+# pylint: disable=unused-import
+from tensorflow.python import tf2
+from tensorflow.python.util.tf_export import keras_export
+
 __version__ = "2.10.0"
 
 keras_export("keras.__version__").export_constant(__name__, "__version__")

keras/activations.py

@@ -17,12 +17,14 @@
 import sys
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 import keras.layers.activation as activation_layers
 from keras import backend
 from keras.utils import generic_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 # b/123041942
 # In TF 2.x, if the `tf.nn.softmax` is used as an activation function in Keras
 # layers, it gets serialized as 'softmax_v2' instead of 'softmax' as the

keras/api/create_python_api_wrapper.py

@@ -23,11 +23,12 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+import keras  # pylint: disable=unused-import
+
+# isort: off
 from tensorflow.python.tools.api.generator import (
     create_python_api,
 )
 
-import keras  # pylint: disable=unused-import
-
 if __name__ == "__main__":
     create_python_api.main()

keras/api/tests/api_compatibility_test.py

@@ -34,6 +34,8 @@ import sys
 
 import six
 import tensorflow as tf
+
+# isort: off
 from google.protobuf import message
 from google.protobuf import text_format
 from tensorflow.python.lib.io import file_io

keras/applications/convnext.py

@@ -26,7 +26,6 @@ References:
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import layers
@@ -35,6 +34,9 @@ from keras.applications import imagenet_utils
 from keras.engine import sequential
 from keras.engine import training as training_lib
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/convnext/"
 )

keras/applications/densenet.py

@@ -21,7 +21,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -30,6 +29,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/" "keras-applications/densenet/"
 )

keras/applications/efficientnet.py

@@ -25,7 +25,6 @@ import copy
 import math
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -34,6 +33,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = "https://storage.googleapis.com/keras-applications/"
 
 WEIGHTS_HASHES = {

keras/applications/efficientnet_v2.py

@@ -25,7 +25,6 @@ import copy
 import math
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import layers
@@ -34,6 +33,9 @@ from keras.engine import training
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = "https://storage.googleapis.com/tensorflow/keras-applications/efficientnet_v2/"  # noqa: E501
 
 WEIGHTS_HASHES = {

keras/applications/imagenet_utils.py

@@ -18,12 +18,14 @@ import json
 import warnings
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import activations
 from keras import backend
 from keras.utils import data_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 CLASS_INDEX = None
 CLASS_INDEX_PATH = (
     "https://storage.googleapis.com/download.tensorflow.org/"

keras/applications/inception_resnet_v2.py

@@ -22,7 +22,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -31,6 +30,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHT_URL = (
     "https://storage.googleapis.com/tensorflow/"
     "keras-applications/inception_resnet_v2/"

keras/applications/inception_v3.py

@@ -21,7 +21,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -30,6 +29,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/"
     "inception_v3/inception_v3_weights_tf_dim_ordering_tf_kernels.h5"

keras/applications/mobilenet.py

@@ -62,8 +62,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -72,6 +70,10 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHT_PATH = (
     "https://storage.googleapis.com/tensorflow/" "keras-applications/mobilenet/"
 )

keras/applications/mobilenet_v2.py

@@ -75,8 +75,6 @@ MACs stands for Multiply Adds
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -85,6 +83,10 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHT_PATH = (
     "https://storage.googleapis.com/tensorflow/"
     "keras-applications/mobilenet_v2/"

keras/applications/mobilenet_v3.py

@@ -17,8 +17,6 @@
 """MobileNet v3 models for Keras."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import models
@@ -27,6 +25,10 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 # TODO(scottzhu): Change this to the GCS path.
 BASE_WEIGHT_PATH = (
     "https://storage.googleapis.com/tensorflow/"

keras/applications/nasnet.py

@@ -39,8 +39,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -49,6 +47,10 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/" "keras-applications/nasnet/"
 )

keras/applications/regnet.py

@@ -27,7 +27,6 @@ References:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import layers
@@ -36,6 +35,9 @@ from keras.engine import training
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/regnet/"
 )

keras/applications/resnet.py

@@ -21,7 +21,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -30,6 +29,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/resnet/"
 )

keras/applications/resnet_rs.py

@@ -27,7 +27,6 @@ from typing import List
 from typing import Union
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import layers
@@ -36,6 +35,9 @@ from keras.engine import training
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 BASE_WEIGHTS_URL = (
     "https://storage.googleapis.com/tensorflow/" "keras-applications/resnet_rs/"
 )

keras/applications/resnet_v2.py

@@ -20,11 +20,12 @@ Reference:
     (https://arxiv.org/abs/1603.05027) (CVPR 2016)
 """
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras.applications import imagenet_utils
 from keras.applications import resnet
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export(
     "keras.applications.resnet_v2.ResNet50V2", "keras.applications.ResNet50V2"

keras/applications/vgg16.py

@@ -21,7 +21,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -30,6 +29,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/"
     "vgg16/vgg16_weights_tf_dim_ordering_tf_kernels.h5"

keras/applications/vgg19.py

@@ -21,7 +21,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -30,6 +29,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/"
     "vgg19/vgg19_weights_tf_dim_ordering_tf_kernels.h5"

keras/applications/xception.py

@@ -24,7 +24,6 @@ Reference:
 """
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.applications import imagenet_utils
@@ -33,6 +32,9 @@ from keras.layers import VersionAwareLayers
 from keras.utils import data_utils
 from keras.utils import layer_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 TF_WEIGHTS_PATH = (
     "https://storage.googleapis.com/tensorflow/keras-applications/"
     "xception/xception_weights_tf_dim_ordering_tf_kernels.h5"

keras/backend.py

@@ -32,13 +32,6 @@ import weakref
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.core.protobuf import config_pb2
-from tensorflow.python.eager import context
-from tensorflow.python.eager.context import get_config
-from tensorflow.python.framework import config
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend_config
 from keras.distribute import distribute_coordinator_utils as dc
@@ -48,6 +41,15 @@ from keras.utils import object_identity
 from keras.utils import tf_contextlib
 from keras.utils import tf_inspect
 
+# isort: off
+from tensorflow.core.protobuf import config_pb2
+from tensorflow.python.eager import context
+from tensorflow.python.eager.context import get_config
+from tensorflow.python.framework import config
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 py_all = all
 py_sum = sum
 py_any = any

keras/backend_config.py

@@ -15,6 +15,8 @@
 """Keras backend config API."""
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.util.tf_export import keras_export
 
 # The type of float to use throughout a session.

keras/backend_test.py

@@ -21,11 +21,6 @@ import numpy as np
 import scipy.sparse
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.eager import context
-from tensorflow.python.eager.context import get_config
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 from keras import activations
 from keras import backend
@@ -36,6 +31,13 @@ from keras.testing_infra import test_combinations
 from keras.utils import tf_inspect
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.eager import context
+from tensorflow.python.eager.context import get_config
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 def compare_single_input_op_to_numpy(
     keras_op,

keras/benchmarks/eager_microbenchmarks_test.py

@@ -17,11 +17,13 @@
 import time
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.eager import context
-from tensorflow.python.eager.context import get_executor
 
 from keras.utils import tf_inspect
 
+# isort: off
+from tensorflow.python.eager import context
+from tensorflow.python.eager.context import get_executor
+
 
 def _run_benchmark(func, num_iters, execution_mode=None):
     with context.execution_mode(execution_mode):

keras/benchmarks/model_components_benchmarks_test.py

@@ -18,6 +18,8 @@ import time
 
 import numpy as np
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.eager import context
 from tensorflow.python.eager.context import get_executor
 

keras/benchmarks/optimizer_benchmarks_test.py

@@ -15,13 +15,15 @@
 """Benchmark tests for Keras optimizers."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform.benchmark import (
-    ParameterizedBenchmark,
-)
 
 from keras.benchmarks import benchmark_util
 from keras.optimizers.optimizer_v2 import adam
 
+# isort: off
+from tensorflow.python.platform.benchmark import (
+    ParameterizedBenchmark,
+)
+
 
 def bidirect_imdb_lstm_config():
     """Bidirectional LSTM model and IMDB data."""

keras/callbacks.py

@@ -27,10 +27,6 @@ import time
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util import deprecation
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras.distribute import distributed_file_utils
@@ -44,6 +40,12 @@ from keras.utils.data_utils import Sequence
 from keras.utils.generic_utils import Progbar
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util import deprecation
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 try:
     import requests
 except ImportError:
@@ -2523,11 +2525,12 @@ class TensorBoard(Callback, version_utils.TensorBoardVersionSelector):
     def _configure_embeddings(self):
         """Configure the Projector for embeddings."""
         # TODO(omalleyt): Add integration tests.
-        from google.protobuf import text_format
-
         from keras.layers import core
         from keras.protobuf import projector_config_pb2
 
+        # isort: off
+        from google.protobuf import text_format
+
         config = projector_config_pb2.ProjectorConfig()
         for layer in self.model.layers:
             if isinstance(layer, core.Embedding):

keras/callbacks_test.py

@@ -30,7 +30,6 @@ from unittest import mock
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.platform import tf_logging as logging
 
 import keras
 from keras.callbacks import BackupAndRestore
@@ -45,6 +44,9 @@ from keras.testing_infra import test_utils
 from keras.utils import io_utils
 from keras.utils import np_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 try:
     import h5py  # pylint:disable=g-import-not-at-top
 except ImportError:

keras/callbacks_v1.py

@@ -20,12 +20,14 @@ import os
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras import callbacks
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export(v1=["keras.callbacks.TensorBoard"])
 class TensorBoard(callbacks.TensorBoard):
@@ -315,6 +317,7 @@ class TensorBoard(callbacks.TensorBoard):
                 embeddings_metadata = self.embeddings_metadata
 
             try:
+                # isort: off
                 from tensorboard.plugins import projector
             except ImportError:
                 raise ImportError(

keras/constraints.py

@@ -17,13 +17,15 @@
 """Constraints: functions that impose constraints on weight values."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras.utils.generic_utils import deserialize_keras_object
 from keras.utils.generic_utils import serialize_keras_object
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 
 @keras_export("keras.constraints.Constraint")
 class Constraint:

keras/datasets/boston_housing.py

@@ -15,10 +15,12 @@
 """Boston housing price regression dataset."""
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.boston_housing.load_data")
 def load_data(path="boston_housing.npz", test_split=0.2, seed=113):

keras/datasets/cifar10.py

@@ -17,12 +17,14 @@
 import os
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.datasets.cifar import load_batch
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.cifar10.load_data")
 def load_data():

keras/datasets/cifar100.py

@@ -17,12 +17,14 @@
 import os
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.datasets.cifar import load_batch
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.cifar100.load_data")
 def load_data(label_mode="fine"):

keras/datasets/fashion_mnist.py

@@ -18,10 +18,12 @@ import gzip
 import os
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.fashion_mnist.load_data")
 def load_data():

keras/datasets/imdb.py

@@ -17,12 +17,14 @@
 import json
 
 import numpy as np
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.preprocessing.sequence import _remove_long_seq
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.imdb.load_data")
 def load_data(

keras/datasets/mnist.py

@@ -15,10 +15,12 @@
 """MNIST handwritten digits dataset."""
 
 import numpy as np
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.mnist.load_data")
 def load_data(path="mnist.npz"):

keras/datasets/reuters.py

@@ -17,12 +17,14 @@
 import json
 
 import numpy as np
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.preprocessing.sequence import _remove_long_seq
 from keras.utils.data_utils import get_file
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.datasets.reuters.load_data")
 def load_data(

keras/distribute/ctl_correctness_test.py

@@ -17,7 +17,6 @@
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.ops.losses import losses_impl
 
 import keras
 from keras import optimizers
@@ -27,6 +26,9 @@ from keras.distribute import optimizer_combinations
 from keras.distribute import strategy_combinations
 from keras.testing_infra import test_utils
 
+# isort: off
+from tensorflow.python.ops.losses import losses_impl
+
 _NUM_SAMPLES = 66
 _BATCH_SIZE = 32
 _RANDOM_SEED = 1337

keras/distribute/custom_training_loop_metrics_test.py

@@ -17,13 +17,15 @@
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 from keras import metrics
 from keras.distribute import strategy_combinations
 
+# isort: off
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 class KerasMetricsTest(tf.test.TestCase, parameterized.TestCase):
     @tf.__internal__.distribute.combinations.generate(

keras/distribute/custom_training_loop_optimizer_test.py

@@ -16,13 +16,15 @@
 
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.distribute import values
 
 from keras.distribute import (
     strategy_combinations as keras_strategy_combinations,
 )
 from keras.optimizers.optimizer_v2 import gradient_descent
 
+# isort: off
+from tensorflow.python.distribute import values
+
 
 class OptimizerTest(tf.test.TestCase, parameterized.TestCase):
     @tf.__internal__.distribute.combinations.generate(

keras/distribute/dataset_creator_model_fit_test.py

@@ -16,15 +16,17 @@
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 from keras.distribute import dataset_creator_model_fit_test_base as test_base
 from keras.distribute import strategy_combinations
 from keras.testing_infra import test_utils
 from keras.utils import dataset_creator
 
+# isort: off
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 # TODO(rchao): Investigate why there cannot be single worker and multi worker
 # PS strategies running in the same shard.

keras/distribute/dataset_creator_model_fit_test_base.py

@@ -19,7 +19,6 @@ import os
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.platform import tf_logging as logging
 
 import keras
 from keras import callbacks as callbacks_lib
@@ -29,6 +28,9 @@ from keras.layers.preprocessing import string_lookup
 from keras.optimizers.optimizer_v2 import gradient_descent
 from keras.utils import dataset_creator
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 
 class DatasetCreatorModelFitTestBase(tf.test.TestCase, parameterized.TestCase):
     """The base class for DatasetCreator with Model.fit tests."""

keras/distribute/distribute_coordinator_utils.py

@@ -32,6 +32,8 @@ import threading
 import time
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.core.protobuf import cluster_pb2
 from tensorflow.python.platform import tf_logging as logging
 

keras/distribute/distribute_strategy_test.py

@@ -19,9 +19,6 @@ import os
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.distribute.cluster_resolver import (
-    SimpleClusterResolver,
-)
 
 import keras
 from keras import backend
@@ -47,6 +44,11 @@ from keras.testing_infra import test_utils
 from keras.utils import losses_utils
 from keras.utils import np_utils
 
+# isort: off
+from tensorflow.python.distribute.cluster_resolver import (
+    SimpleClusterResolver,
+)
+
 _RANDOM_SEED = 1337
 _TRAIN_SIZE = 200
 _INPUT_SIZE = (10,)

keras/distribute/distributed_training_utils_v1.py

@@ -18,7 +18,6 @@ import functools
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import callbacks
@@ -31,6 +30,9 @@ from keras.optimizers.optimizer_v2 import optimizer_v2
 from keras.utils import tf_contextlib
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 # pylint:disable=protected-access
 
 

keras/distribute/mirrored_strategy_test.py

@@ -17,10 +17,6 @@
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.eager import backprop
-from tensorflow.python.training import (
-    optimizer as optimizer_lib,
-)
 
 import keras
 from keras.engine import training as keras_training
@@ -28,6 +24,12 @@ from keras.layers import core as keras_core
 from keras.optimizers.optimizer_v2 import rmsprop
 from keras.utils import kpl_test_utils
 
+# isort: off
+from tensorflow.python.eager import backprop
+from tensorflow.python.training import (
+    optimizer as optimizer_lib,
+)
+
 
 class MiniModel(keras_training.Model):
     """Minimal model for mnist.

keras/distribute/multi_worker_testing_utils.py

@@ -18,6 +18,11 @@ import threading
 import unittest
 
 import tensorflow.compat.v2 as tf
+
+import keras
+from keras.optimizers.optimizer_v2 import gradient_descent
+
+# isort: off
 from tensorflow.python.distribute.cluster_resolver import (
     SimpleClusterResolver,
 )
@@ -26,9 +31,6 @@ from tensorflow.python.training.server_lib import (
     ClusterSpec,
 )
 
-import keras
-from keras.optimizers.optimizer_v2 import gradient_descent
-
 _portpicker_import_error = None
 try:
     import portpicker  # pylint: disable=g-import-not-at-top

keras/distribute/parameter_server_evaluation_test.py

@@ -17,6 +17,11 @@
 import time
 
 import tensorflow.compat.v2 as tf
+
+import keras
+from keras.testing_infra import test_utils
+
+# isort: off
 from tensorflow.python.distribute import (
     multi_worker_test_base,
 )
@@ -25,9 +30,6 @@ from tensorflow.python.distribute.cluster_resolver import (
 )
 from tensorflow.python.ops import resource_variable_ops
 
-import keras
-from keras.testing_infra import test_utils
-
 
 # TODO(yuefengz): move the following implementation to Keras core.
 class MeanMetricSpec(tf.TypeSpec):

keras/distribute/sidecar_evaluator.py

@@ -15,6 +15,8 @@
 """Python module for evaluation loop."""
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.platform import tf_logging as logging
 from tensorflow.python.util import deprecation
 from tensorflow.python.util.tf_export import keras_export

keras/distribute/sidecar_evaluator_test.py

@@ -22,13 +22,15 @@ import time
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.platform import tf_logging as logging
 
 import keras
 from keras.distribute import sidecar_evaluator as sidecar_evaluator_lib
 from keras.optimizers.optimizer_v2 import gradient_descent
 from keras.testing_infra import test_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 _BATCH_SIZE = 32
 
 

keras/dtensor/layout_map.py

@@ -19,13 +19,14 @@ import contextlib
 import re
 import threading
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras.dtensor import dtensor_api as dtensor
 from keras.dtensor import lazy_variable
 from keras.dtensor import utils
 from keras.engine import base_layer
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 # pylint: disable=missing-class-docstring
 
 # We will skip the path for certain attributes when mapping the layout, e.g.

keras/dtensor/lazy_variable.py

@@ -16,6 +16,7 @@
 
 import threading
 
+# isort: off
 from tensorflow.core.framework import attr_value_pb2
 from tensorflow.python.eager import context
 from tensorflow.python.framework import ops

keras/dtensor/optimizers.py

@@ -15,8 +15,6 @@
 """DTensor specific Keras optimizers."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras.dtensor import dtensor_api as dtensor
 from keras.optimizers.optimizer_experimental import adadelta
@@ -27,6 +25,10 @@ from keras.optimizers.optimizer_experimental import rmsprop
 from keras.optimizers.optimizer_experimental import sgd
 from keras.optimizers.schedules import learning_rate_schedule
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 
 # pylint: disable=protected-access,missing-class-docstring
 class Optimizer(optimizer_lib._BaseOptimizer):

keras/dtensor/test_util.py

@@ -17,6 +17,8 @@
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
+
+# isort: off
 from tensorflow.dtensor.python import api as dtensor_api
 from tensorflow.python.eager import context
 

keras/engine/base_layer.py

@@ -28,13 +28,6 @@ import weakref
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from google.protobuf import json_format
-from tensorflow.python.platform import tf_logging
-from tensorflow.python.util.tf_export import (
-    get_canonical_name_for_symbol,
-)
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras import constraints
@@ -65,6 +58,15 @@ from keras.utils.tf_utils import (
     is_tensor_or_tensor_list,  # pylint: disable=unused-import
 )
 
+# isort: off
+from google.protobuf import json_format
+from tensorflow.python.platform import tf_logging
+from tensorflow.python.util.tf_export import (
+    get_canonical_name_for_symbol,
+)
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 # pylint: disable=g-inconsistent-quotes
 metrics_mod = generic_utils.LazyLoader(
     "metrics_mod", globals(), "keras.metrics"

keras/engine/base_layer_utils.py

@@ -14,14 +14,11 @@
 # ==============================================================================
 """Contains private utilities used mainly by the base Layer class."""
 
-import tensorflow.compat.v2 as tf
-import tensorflow.compat.v1 as tf1
-
 import functools
 import threading
 
+import tensorflow.compat.v1 as tf1
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.dtensor import dtensor_api as dtensor
@@ -29,6 +26,9 @@ from keras.utils import control_flow_util
 from keras.utils import tf_inspect
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 _call_context = threading.local()
 
 

keras/engine/base_layer_v1.py

@@ -22,8 +22,6 @@ import threading
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras import constraints
@@ -50,6 +48,10 @@ from keras.utils.tf_utils import (
     is_tensor_or_tensor_list,  # pylint: disable=unused-import
 )
 
+# isort: off
+from tensorflow.python.platform import tf_logging
+from tensorflow.tools.docs import doc_controls
+
 
 # pylint: disable=g-classes-have-attributes
 class Layer(base_layer.Layer):

keras/engine/base_preprocessing_layer.py

@@ -17,14 +17,16 @@
 import abc
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.eager import context
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras.engine import data_adapter
 from keras.engine.base_layer import Layer
 from keras.utils import version_utils
 
+# isort: off
+from tensorflow.python.eager import context
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 keras_kpl_gauge = tf.__internal__.monitoring.BoolGauge(
     "/tensorflow/api/keras/layers/preprocessing",
     "keras preprocessing layers usage",

keras/engine/data_adapter.py

@@ -23,6 +23,14 @@ import random
 
 import numpy as np
 import tensorflow.compat.v2 as tf
+
+from keras import backend
+from keras.engine import training_utils
+from keras.utils import data_utils
+from keras.utils import dataset_creator
+from keras.utils import tf_utils
+
+# isort: off
 from tensorflow.python.distribute.input_lib import (
     DistributedDataset,
 )
@@ -31,12 +39,6 @@ from tensorflow.python.framework import type_spec
 from tensorflow.python.platform import tf_logging as logging
 from tensorflow.python.util.tf_export import keras_export
 
-from keras import backend
-from keras.engine import training_utils
-from keras.utils import data_utils
-from keras.utils import dataset_creator
-from keras.utils import tf_utils
-
 try:
     import pandas as pd  # pylint: disable=g-import-not-at-top
 except ImportError:

keras/engine/data_adapter_test.py

@@ -19,7 +19,6 @@ import math
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.eager import context
 
 import keras
 from keras.engine import data_adapter
@@ -27,6 +26,9 @@ from keras.testing_infra import test_combinations
 from keras.testing_infra import test_utils
 from keras.utils import data_utils
 
+# isort: off
+from tensorflow.python.eager import context
+
 
 class DummyArrayLike:
     """Dummy array-like object."""

keras/engine/functional.py

@@ -22,8 +22,6 @@ import itertools
 import warnings
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras.dtensor import layout_map as layout_map_lib
@@ -41,6 +39,10 @@ from keras.utils import generic_utils
 from keras.utils import tf_inspect
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.tools.docs import doc_controls
+
 
 # pylint: disable=g-classes-have-attributes
 class Functional(training_lib.Model):

keras/engine/functional_test.py

@@ -18,10 +18,6 @@ import warnings
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.framework import extension_type
-from tensorflow.python.training.tracking.util import (
-    Checkpoint,
-)
 
 from keras import backend
 from keras import layers
@@ -37,6 +33,12 @@ from keras.testing_infra import test_utils
 from keras.utils import layer_utils
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.framework import extension_type
+from tensorflow.python.training.tracking.util import (
+    Checkpoint,
+)
+
 
 class NetworkConstructionTest(test_combinations.TestCase):
     def test_default_model_name(self):

keras/engine/input_layer.py

@@ -16,7 +16,6 @@
 """Input layer code (`Input` and `InputLayer`)."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.distribute import distributed_training_utils
@@ -27,6 +26,9 @@ from keras.saving.saved_model import layer_serialization
 from keras.utils import tf_utils
 from keras.utils import traceback_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 def _assert_other_arg_none(arg_name, arg):
     if arg is not None:

keras/engine/input_layer_test.py

@@ -15,7 +15,6 @@
 """Tests for InputLayer construction."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.framework import type_spec
 
 from keras import backend
 from keras.engine import functional
@@ -24,6 +23,9 @@ from keras.layers import core
 from keras.saving import model_config
 from keras.testing_infra import test_combinations
 
+# isort: off
+from tensorflow.python.framework import type_spec
+
 
 class TwoTensors(tf.__internal__.CompositeTensor):
     """A simple value type to test TypeSpec.

keras/engine/input_spec.py

@@ -17,11 +17,13 @@
 """Contains the InputSpec class."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.python.util.tf_export import tf_export
 
 from keras import backend
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.python.util.tf_export import tf_export
+
 
 @keras_export(
     "keras.layers.InputSpec",

keras/engine/keras_tensor.py

@@ -15,10 +15,12 @@
 """Keras Input Tensor used to track functional API Topology."""
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.data.util import structure
 
 from keras.utils import object_identity
 
+# isort: off
+from tensorflow.python.data.util import structure
+
 # pylint: disable=g-classes-have-attributes
 
 

keras/engine/sequential.py

@@ -18,8 +18,6 @@
 import copy
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import layers as layer_module
 from keras.engine import base_layer
@@ -33,6 +31,10 @@ from keras.utils import tf_inspect
 from keras.utils import tf_utils
 from keras.utils import traceback_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+
 SINGLE_LAYER_OUTPUT_ERROR_MSG = (
     "All layers in a Sequential model should have "
     "a single output tensor. For multi-output "

keras/engine/sequential_test.py

@@ -17,14 +17,16 @@
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 import keras
 from keras.testing_infra import test_combinations
 from keras.testing_infra import test_utils
 
+# isort: off
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 class TestSequential(test_combinations.TestCase):
     """Most Sequential model API tests are covered in `training_test.py`."""

keras/engine/training.py

@@ -23,10 +23,6 @@ import weakref
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.eager import context
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.util.tf_export import keras_export
-from tensorflow.tools.docs import doc_controls
 
 from keras import backend
 from keras import callbacks as callbacks_module
@@ -58,6 +54,12 @@ from keras.utils import traceback_utils
 from keras.utils import version_utils
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.eager import context
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.util.tf_export import keras_export
+from tensorflow.tools.docs import doc_controls
+
 try:
     import h5py
 except ImportError:

keras/engine/training_arrays_test.py

@@ -21,9 +21,6 @@ from unittest import mock
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 import keras
 from keras.engine import data_adapter
@@ -32,6 +29,11 @@ from keras.testing_infra import test_combinations
 from keras.testing_infra import test_utils
 from keras.utils import io_utils
 
+# isort: off
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 def _create_dataset(num_samples, batch_size):
     input_data = np.random.rand(num_samples, 1)

keras/engine/training_arrays_v1.py

@@ -18,7 +18,6 @@ import functools
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import callbacks as cbks
@@ -29,6 +28,9 @@ from keras.utils.generic_utils import make_batches
 from keras.utils.generic_utils import slice_arrays
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 # pylint: disable=protected-access
 
 

keras/engine/training_dataset_test.py

@@ -19,7 +19,6 @@ import sys
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 import keras
 from keras import callbacks
@@ -28,6 +27,9 @@ from keras.testing_infra import test_combinations
 from keras.testing_infra import test_utils
 from keras.utils import io_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 
 class BatchCounterCallback(callbacks.Callback):
     def __init__(self):

keras/engine/training_distributed_v1.py

@@ -16,8 +16,6 @@
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.distribute import input_lib
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import callbacks as cbks
@@ -29,6 +27,10 @@ from keras.engine import training_utils_v1
 from keras.utils.generic_utils import Progbar
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.distribute import input_lib
+from tensorflow.python.platform import tf_logging as logging
+
 # pylint: disable=protected-access
 
 

keras/engine/training_eager_v1.py

@@ -16,8 +16,6 @@
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.eager.backprop import GradientTape
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras.engine import training_utils
@@ -25,6 +23,10 @@ from keras.engine import training_utils_v1
 from keras.mixed_precision import loss_scale_optimizer
 from keras.utils import losses_utils
 
+# isort: off
+from tensorflow.python.eager.backprop import GradientTape
+from tensorflow.python.platform import tf_logging as logging
+
 # pylint: disable=protected-access
 
 

keras/engine/training_generator_v1.py

@@ -20,7 +20,6 @@ import math
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import callbacks as cbks
@@ -30,6 +29,9 @@ from keras.utils import data_utils
 from keras.utils import generic_utils
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 # pylint: disable=protected-access
 
 

keras/engine/training_test.py

@@ -23,13 +23,6 @@ import tempfile
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
-from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.training.rmsprop import (
-    RMSPropOptimizer,
-)
 
 import keras
 from keras import backend
@@ -51,6 +44,15 @@ from keras.utils import data_utils
 from keras.utils import io_utils
 from keras.utils import np_utils
 
+# isort: off
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+from tensorflow.python.platform import tf_logging as logging
+from tensorflow.python.training.rmsprop import (
+    RMSPropOptimizer,
+)
+
 try:
     import scipy.sparse as scipy_sparse  # pylint: disable=g-import-not-at-top
 except ImportError:

keras/engine/training_utils_v1.py

@@ -24,7 +24,6 @@ import time
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import callbacks as cbks
@@ -35,6 +34,9 @@ from keras.utils import generic_utils
 from keras.utils import losses_utils
 from keras.utils import tf_inspect
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 
 def is_composite_or_composite_value(tensor):
     """Returns true if 'tensor' is a CompositeTensor or a CT Value object."""

keras/engine/training_utils_v1_test.py

@@ -21,7 +21,6 @@ import time
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras.engine import keras_tensor
@@ -29,6 +28,9 @@ from keras.engine import training_utils_v1
 from keras.testing_infra import test_combinations
 from keras.testing_infra import test_utils
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 
 class ModelInputsTest(tf.test.TestCase):
     def test_single_thing(self):

keras/engine/training_v1.py

@@ -20,7 +20,6 @@ import numpy as np
 
 # pylint: disable=g-classes-have-attributes
 import tensorflow.compat.v2 as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from keras import backend
 from keras import losses
@@ -48,6 +47,9 @@ from keras.utils import tf_inspect
 from keras.utils import tf_utils
 from keras.utils.mode_keys import ModeKeys
 
+# isort: off
+from tensorflow.python.platform import tf_logging as logging
+
 try:
     from scipy.sparse import issparse  # pylint: disable=g-import-not-at-top
 except ImportError:

keras/estimator/__init__.py

@@ -15,6 +15,8 @@
 """Keras estimator API."""
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.util.tf_export import keras_export
 
 # Keras has undeclared dependency on tensorflow/estimator:estimator_py.
@@ -164,6 +166,7 @@ def model_to_estimator(
     """
 
     try:
+        # isort: off
         from tensorflow_estimator.python.estimator import (
             keras_lib,  # pylint: disable=g-import-not-at-top
         )
@@ -362,6 +365,7 @@ def model_to_estimator_v2(
     """
 
     try:
+        # isort: off
         from tensorflow_estimator.python.estimator import (
             keras_lib,  # pylint: disable=g-import-not-at-top
         )

keras/feature_column/dense_features.py

@@ -21,12 +21,14 @@ from __future__ import print_function
 import json
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.feature_column import base_feature_layer as kfc
 from keras.saving.saved_model import json_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export(v1=["keras.layers.DenseFeatures"])
 class DenseFeatures(kfc._BaseFeaturesLayer):  # pylint: disable=protected-access

keras/feature_column/dense_features_test.py

@@ -21,14 +21,16 @@ from __future__ import print_function
 import numpy as np
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
+
+from keras.feature_column import dense_features as df
+from keras.testing_infra import test_combinations
+
+# isort: off
 from tensorflow.python.eager import backprop
 from tensorflow.python.framework import (
     test_util as tf_test_utils,
 )
 
-from keras.feature_column import dense_features as df
-from keras.testing_infra import test_combinations
-
 
 def _initialized_session(config=None):
     sess = tf.compat.v1.Session(config=config)

keras/feature_column/dense_features_v2.py

@@ -19,12 +19,14 @@ from __future__ import division
 from __future__ import print_function
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.feature_column import base_feature_layer as kfc
 from keras.feature_column import dense_features
 from keras.utils import tf_contextlib
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.layers.DenseFeatures", v1=[])
 class DenseFeatures(dense_features.DenseFeatures):

keras/feature_column/dense_features_v2_test.py

@@ -20,11 +20,13 @@ from __future__ import print_function
 
 import numpy as np
 import tensorflow.compat.v2 as tf
-from tensorflow.python.eager import backprop
 
 from keras.feature_column import dense_features_v2 as df
 from keras.testing_infra import test_combinations
 
+# isort: off
+from tensorflow.python.eager import backprop
+
 
 def _initialized_session(config=None):
     sess = tf.compat.v1.Session(config=config)

keras/feature_column/sequence_feature_column.py

@@ -22,11 +22,13 @@ from __future__ import division
 from __future__ import print_function
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.feature_column import base_feature_layer as kfc
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 # pylint: disable=protected-access
 
 

keras/feature_column/sequence_feature_column_integration_test.py

@@ -19,12 +19,6 @@ from __future__ import division
 from __future__ import print_function
 
 import tensorflow.compat.v2 as tf
-from google.protobuf import text_format
-from tensorflow.core.example import example_pb2
-from tensorflow.core.example import feature_pb2
-from tensorflow.python.framework import (
-    test_util as tf_test_utils,
-)
 
 from keras import backend
 from keras.feature_column import dense_features
@@ -33,6 +27,14 @@ from keras.layers import merging
 from keras.layers.rnn import base_rnn
 from keras.layers.rnn import simple_rnn
 
+# isort: off
+from google.protobuf import text_format
+from tensorflow.core.example import example_pb2
+from tensorflow.core.example import feature_pb2
+from tensorflow.python.framework import (
+    test_util as tf_test_utils,
+)
+
 
 class SequenceFeatureColumnIntegrationTest(tf.test.TestCase):
     def _make_sequence_example(self):

keras/initializers/__init__.py

@@ -17,15 +17,17 @@
 import threading
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python import tf2
-from tensorflow.python.ops import init_ops
-from tensorflow.python.util.tf_export import keras_export
 
 from keras.initializers import initializers_v1
 from keras.initializers import initializers_v2
 from keras.utils import generic_utils
 from keras.utils import tf_inspect as inspect
 
+# isort: off
+from tensorflow.python import tf2
+from tensorflow.python.ops import init_ops
+from tensorflow.python.util.tf_export import keras_export
+
 # LOCAL.ALL_OBJECTS is meant to be a global mutable. Hence we need to make it
 # thread-local to avoid concurrent mutations.
 LOCAL = threading.local()

keras/initializers/initializers_v1.py

@@ -16,6 +16,8 @@
 # pylint:disable=g-classes-have-attributes
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.util.tf_export import keras_export
 
 _v1_zeros_initializer = tf.compat.v1.zeros_initializer

keras/initializers/initializers_v2.py

@@ -17,11 +17,13 @@
 import math
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.dtensor import utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 _PARTITION_SHAPE = "partition_shape"
 _PARTITION_OFFSET = "partition_offset"
 _LAYOUT = "layout"

keras/integration_test/central_storage_strategy_test.py

@@ -16,6 +16,8 @@
 
 import tensorflow.compat.v2 as tf
 from absl.testing import parameterized
+
+# isort: off
 from tensorflow.python.distribute import (
     combinations as ds_combinations,
 )

keras/integration_test/gradient_checkpoint_test.py

@@ -16,6 +16,8 @@
 import gc
 
 import tensorflow.compat.v2 as tf
+
+# isort: off
 from tensorflow.python.framework import (
     test_util as tf_test_utils,
 )

keras/integration_test/tpu_strategy_test.py

@@ -19,6 +19,8 @@ import tempfile
 
 import tensorflow.compat.v2 as tf
 from absl import flags
+
+# isort: off
 from tensorflow.python.framework import (
     test_util as tf_test_utils,
 )

keras/layers/__init__.py

@@ -16,8 +16,6 @@
 
 import tensorflow.compat.v2 as tf
 
-from tensorflow.python import tf2
-
 from keras.engine.base_layer import Layer
 from keras.engine.base_preprocessing_layer import PreprocessingLayer
 
@@ -157,6 +155,9 @@ from keras.layers.reshaping.zero_padding1d import ZeroPadding1D
 from keras.layers.reshaping.zero_padding2d import ZeroPadding2D
 from keras.layers.reshaping.zero_padding3d import ZeroPadding3D
 
+# isort: off
+from tensorflow.python import tf2
+
 if tf.__internal__.tf2.enabled():
     from keras.layers.normalization.batch_normalization import (
         BatchNormalization,

keras/layers/activation/elu.py

@@ -15,12 +15,13 @@
 """Exponential Linear Unit activation layer."""
 # pylint: disable=g-classes-have-attributes,g-direct-tensorflow-import
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras import backend
 from keras.engine.base_layer import Layer
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.layers.ELU")
 class ELU(Layer):

keras/layers/activation/leaky_relu.py

@@ -15,12 +15,13 @@
 """Leaky version of a Rectified Linear Unit activation layer."""
 # pylint: disable=g-classes-have-attributes,g-direct-tensorflow-import
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras import backend
 from keras.engine.base_layer import Layer
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.layers.LeakyReLU")
 class LeakyReLU(Layer):

keras/layers/activation/prelu.py

@@ -15,8 +15,6 @@
 """Parametric Rectified Linear Unit activation layer."""
 # pylint: disable=g-classes-have-attributes,g-direct-tensorflow-import
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras import backend
 from keras import constraints
 from keras import initializers
@@ -25,6 +23,9 @@ from keras.engine.base_layer import Layer
 from keras.engine.input_spec import InputSpec
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.layers.PReLU")
 class PReLU(Layer):

keras/layers/activation/relu.py

@@ -15,12 +15,13 @@
 """Rectified Linear Unit activation layer."""
 # pylint: disable=g-classes-have-attributes,g-direct-tensorflow-import
 
-from tensorflow.python.util.tf_export import keras_export
-
 from keras import backend
 from keras.engine.base_layer import Layer
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 @keras_export("keras.layers.ReLU")
 class ReLU(Layer):

keras/layers/activation/softmax.py

@@ -16,12 +16,14 @@
 # pylint: disable=g-classes-have-attributes,g-direct-tensorflow-import
 
 import tensorflow.compat.v2 as tf
-from tensorflow.python.util.tf_export import keras_export
 
 from keras import backend
 from keras.engine.base_layer import Layer
 from keras.utils import tf_utils
 
+# isort: off
+from tensorflow.python.util.tf_export import keras_export
+
 
 def _large_compatible_negative(tensor_type):
     """Large negative number as Tensor.