Fixed error message in wrap_tensor_or_dict_of_tensors_in_identity

devidipak · Aug 17, 2018 · 6cc8d30 · 6cc8d30
1 parent 2fb97d4
commit 6cc8d30
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow_model_analysis/eval_saved_model/util.py b/tensorflow_model_analysis/eval_saved_model/util.py
@@ -53,8 +53,7 @@ def _wrap_tensor_in_identity(tensor):
           values=tensor.values,
           dense_shape=tensor.dense_shape)
     else:
-      raise ValueError('could not wrap Tensor %s in identity' %
-                       str(tensor_or_dict_of_tensors))
+      raise ValueError('could not wrap Tensor %s in identity' % str(tensor))
 
   if isinstance(tensor_or_dict_of_tensors, dict):
     result = {}