diff --git a/tensorflow/python/training/checkpoint_ops_test.py b/tensorflow/python/training/checkpoint_ops_test.py index a0fd2dc6bae..5a6a66f1312 100644 --- a/tensorflow/python/training/checkpoint_ops_test.py +++ b/tensorflow/python/training/checkpoint_ops_test.py @@ -18,6 +18,7 @@ from __future__ import division from __future__ import print_function import os + import numpy as np from tensorflow.python.framework import constant_op diff --git a/tensorflow/python/training/checkpoint_utils.py b/tensorflow/python/training/checkpoint_utils.py index 64ba099ce79..1c249035980 100644 --- a/tensorflow/python/training/checkpoint_utils.py +++ b/tensorflow/python/training/checkpoint_utils.py @@ -19,6 +19,7 @@ from __future__ import division from __future__ import print_function import time + import six from tensorflow.python.distribute import distribution_strategy_context diff --git a/tensorflow/python/training/checkpoint_utils_test.py b/tensorflow/python/training/checkpoint_utils_test.py index 59972e6b230..caba001a08b 100644 --- a/tensorflow/python/training/checkpoint_utils_test.py +++ b/tensorflow/python/training/checkpoint_utils_test.py @@ -20,6 +20,7 @@ from __future__ import print_function import os import time + import numpy as np from tensorflow.core.protobuf import config_pb2 diff --git a/tensorflow/python/training/experimental/mixed_precision_test.py b/tensorflow/python/training/experimental/mixed_precision_test.py index bbf9badfcf1..7397ae9a086 100644 --- a/tensorflow/python/training/experimental/mixed_precision_test.py +++ b/tensorflow/python/training/experimental/mixed_precision_test.py @@ -18,6 +18,7 @@ from __future__ import division from __future__ import print_function import os + from absl.testing import parameterized from tensorflow.core.protobuf import config_pb2 diff --git a/tensorflow/python/training/session_manager.py b/tensorflow/python/training/session_manager.py index 9c2db27af2c..fe622097e74 100644 --- a/tensorflow/python/training/session_manager.py +++ b/tensorflow/python/training/session_manager.py @@ -18,6 +18,7 @@ from __future__ import division from __future__ import print_function import time + import numpy as np from tensorflow.python.client import session diff --git a/tensorflow/python/training/warm_starting_util.py b/tensorflow/python/training/warm_starting_util.py index fa334a41875..4d329b4b316 100644 --- a/tensorflow/python/training/warm_starting_util.py +++ b/tensorflow/python/training/warm_starting_util.py @@ -19,6 +19,7 @@ from __future__ import division from __future__ import print_function import collections + import six from tensorflow.python.framework import errors diff --git a/tensorflow/python/training/warm_starting_util_test.py b/tensorflow/python/training/warm_starting_util_test.py index 14fa2430477..084779b332c 100644 --- a/tensorflow/python/training/warm_starting_util_test.py +++ b/tensorflow/python/training/warm_starting_util_test.py @@ -19,6 +19,7 @@ from __future__ import division from __future__ import print_function import os + import numpy as np import six