Better error message when a MetricSpec overspecifies keys.

PiperOrigin-RevId: 158569852
TeeeJay · Jun 9, 2017 · c9275fe · c9275fe
1 parent 386ed6f
commit c9275fe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow/contrib/learn/python/learn/metric_spec.py b/tensorflow/contrib/learn/python/learn/metric_spec.py
@@ -401,7 +401,9 @@ def _get_dict(name, dict_or_tensor, key):
  if not isinstance(dict_or_tensor, dict):
  raise ValueError('MetricSpec with ' + name + '_key specified'
  ' requires ' +
- name + 's dict, got %s' % dict_or_tensor)
+ name + 's dict, got %s.\n' % dict_or_tensor +
+ 'You must not provide a %s_key if you ' % name +
+ 'only have a single Tensor as %ss.' % name)
  if key not in dict_or_tensor:
  raise KeyError(
  'Key \'%s\' missing from %s.' % (key, dict_or_tensor.keys()))