From 6b524a9e75c6b8170453000df89e1ff73848229e Mon Sep 17 00:00:00 2001 From: Adam Cogdell <17283933+BlaziusMaximus@users.noreply.github.com> Date: Mon, 28 Nov 2022 13:52:23 -0800 Subject: [PATCH] Change dependency from saved_model_cli to saved_model_utils (#7039) * Change dependency from saved_model_cli to saved_model_utils * saved_model_utils import change Co-authored-by: Ping Yu <4018+pyu10055@users.noreply.github.com> --- .../tensorflowjs/converters/tf_saved_model_conversion_v2.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py index 9fa80b2759..2d11244941 100644 --- a/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py +++ b/tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py @@ -43,8 +43,8 @@ from tensorflow.python.keras.saving.saving_utils import model_input_signature from tensorflow.python.saved_model.load import load from tensorflow.python.saved_model import loader +from tensorflow.python.tools.saved_model_utils import get_meta_graph_def from tensorflow.python.training.saver import export_meta_graph -from tensorflow.python.tools.saved_model_cli import get_signature_def_map from tensorflow.saved_model.experimental import TrackableResource from google.protobuf.json_format import MessageToDict import tensorflow_hub as hub @@ -640,7 +640,8 @@ def _load_model(saved_model_dir, saved_model_tags): return model def _find_signature(saved_model_dir, saved_model_tags, signature_def): - signature_def_map = get_signature_def_map(saved_model_dir, saved_model_tags) + meta_graph = get_meta_graph_def(saved_model_dir, saved_model_tags) + signature_def_map = meta_graph.signature_def if signature_def not in signature_def_map.keys(): raise ValueError('Signature "%s" does not exist in the saved model' % (signature_def))