diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-18 11:57:04 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-18 12:01:01 -0700 |
| commit | 6a725f6d0974dc71fe4ac311fc8dd16db4257452 (patch) | |
| tree | 79f34c56e8a8ec1ce80705e41b97f93efe8ae493 /tensorflow/contrib/predictor | |
| parent | 1a325f1330cd6e1204d47f1fcf64caf402788477 (diff) | |
Add expected keys to predictor exception if unexpected key detected.
PiperOrigin-RevId: 172634275
Diffstat (limited to 'tensorflow/contrib/predictor')
| -rw-r--r-- | tensorflow/contrib/predictor/predictor.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/predictor/predictor.py b/tensorflow/contrib/predictor/predictor.py index dbc0028259..28fa815684 100644 --- a/tensorflow/contrib/predictor/predictor.py +++ b/tensorflow/contrib/predictor/predictor.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================== - """Abstract base class for all predictors.""" from __future__ import absolute_import @@ -66,8 +65,9 @@ class Predictor(object): expected_keys = set(self.feed_tensors.keys()) unexpected_keys = input_keys - expected_keys if unexpected_keys: - raise ValueError('Got unexpected keys in input_dict: {}'.format( - unexpected_keys)) + raise ValueError( + 'Got unexpected keys in input_dict: {}\nexpected: {}'.format( + unexpected_keys, expected_keys)) feed_dict = {} for key in self.feed_tensors.keys(): |