diff options
Diffstat (limited to 'tensorflow/python/tools/saved_model_cli.py')
| -rw-r--r-- | tensorflow/python/tools/saved_model_cli.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/tools/saved_model_cli.py b/tensorflow/python/tools/saved_model_cli.py index 38fed5335e..9b232865dd 100644 --- a/tensorflow/python/tools/saved_model_cli.py +++ b/tensorflow/python/tools/saved_model_cli.py @@ -308,7 +308,7 @@ def run_saved_model_with_feed_dict(saved_model_dir, tag_set, signature_def_key, # Check if input tensor keys are valid. for input_key_name in input_tensor_key_feed_dict.keys(): - if input_key_name not in inputs_tensor_info.keys(): + if input_key_name not in inputs_tensor_info: raise ValueError( '"%s" is not a valid input key. Please choose from %s, or use ' '--show option.' % |