fix deprecated message.
PiperOrigin-RevId: 316490075
Change-Id: I68d52d80ccf9c2f77f870dd7f4f1d146e1e44a46
diff --git a/tensorflow/python/keras/saving/saving_utils.py b/tensorflow/python/keras/saving/saving_utils.py
index 9a407f6..3c9c335 100644
--- a/tensorflow/python/keras/saving/saving_utils.py
+++ b/tensorflow/python/keras/saving/saving_utils.py
@@ -17,7 +17,6 @@
from __future__ import division
from __future__ import print_function
-import collections
import copy
import os
import six
@@ -31,6 +30,7 @@
from tensorflow.python.keras.utils.io_utils import ask_to_proceed_with_overwrite
from tensorflow.python.platform import tf_logging as logging
from tensorflow.python.util import nest
+from tensorflow.python.util.compat import collections_abc
def extract_model_metrics(model):
@@ -79,7 +79,8 @@
return None
input_specs = _enforce_names_consistency(input_specs)
# Return a list with a single element as the model's input signature.
- if isinstance(input_specs, collections.Sequence) and len(input_specs) == 1:
+ if isinstance(input_specs,
+ collections_abc.Sequence) and len(input_specs) == 1:
# Note that the isinstance check filters out single-element dictionaries,
# which should also be wrapped as a single-element list.
return input_specs
