diff options
| author |  Sergio Guadarrama <sguada@google.com> | 2017-12-13 14:50:24 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-13 14:53:57 -0800 |
| commit | 7fbcd3724379161e379f4693ba66666b6ac4b0aa (patch) | |
| tree | b3c814cdd3c9a8a29960e8dd5a35c5e206430c5b /tensorflow/contrib/slim | |
| parent | 82c1af4a693395489cb43c1bbb7c7be41af654b8 (diff) | |
Avoid modifying items in dataset_data_provider.
PiperOrigin-RevId: 178961790
Diffstat (limited to 'tensorflow/contrib/slim')
| -rw-r--r-- | tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py b/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py index 82c6b5a619..a781c647a1 100644 --- a/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py +++ b/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py @@ -96,12 +96,12 @@ class DatasetDataProvider(data_provider.DataProvider): items = dataset.decoder.list_items() tensors = dataset.decoder.decode(data, items) - if record_key in items: + items_to_tensors = dict(zip(items, tensors)) + if record_key in items_to_tensors: raise ValueError('The item name used for `record_key` cannot also be ' 'used for a dataset item: %s', record_key) - items.append(record_key) - tensors.append(key) + items_to_tensors[record_key] = key super(DatasetDataProvider, self).__init__( - items_to_tensors=dict(zip(items, tensors)), + items_to_tensors=items_to_tensors, num_samples=dataset.num_samples) |