diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-30 00:06:26 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-30 00:09:09 -0700 |
| commit | 28e694db5b549e1ec1e6a7c38fda053c31a87ccb (patch) | |
| tree | 7d7cb9a3c94a7a22938d690341ce96724ae99661 /tensorflow/python/feature_column | |
| parent | 94898251aa7116774f788b5b6c9c9a618c13cea0 (diff) | |
Improve error message when a missing feature name is passed as a unicode string.
PiperOrigin-RevId: 198514621
Diffstat (limited to 'tensorflow/python/feature_column')
| -rw-r--r-- | tensorflow/python/feature_column/feature_column.py | 2 | ||||
| -rw-r--r-- | tensorflow/python/feature_column/feature_column_test.py | 3 |
2 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/python/feature_column/feature_column.py b/tensorflow/python/feature_column/feature_column.py index ffcb9990d5..7aa46af828 100644 --- a/tensorflow/python/feature_column/feature_column.py +++ b/tensorflow/python/feature_column/feature_column.py @@ -2163,7 +2163,7 @@ class _LazyBuilder(object): self._feature_tensors[key] = feature_tensor return feature_tensor - if isinstance(key, str): + if isinstance(key, six.string_types): raise ValueError('Feature {} is not in features dictionary.'.format(key)) if not isinstance(key, _FeatureColumn): diff --git a/tensorflow/python/feature_column/feature_column_test.py b/tensorflow/python/feature_column/feature_column_test.py index f9206f4f38..0af7b9baa9 100644 --- a/tensorflow/python/feature_column/feature_column_test.py +++ b/tensorflow/python/feature_column/feature_column_test.py @@ -137,6 +137,9 @@ class LazyColumnTest(test.TestCase): with self.assertRaisesRegexp(ValueError, 'bbb is not in features dictionary'): builder.get('bbb') + with self.assertRaisesRegexp(ValueError, + 'bbb is not in features dictionary'): + builder.get(u'bbb') def test_not_supported_feature_column(self): |