| 3806 | """An identity `Dataset` that stores options.""" |
| 3807 | |
| 3808 | def __init__(self, input_dataset, options): |
| 3809 | self._input_dataset = input_dataset |
| 3810 | self._options = input_dataset.options() |
| 3811 | if self._options: |
| 3812 | self._options = self._options.merge(options) |
| 3813 | else: |
| 3814 | self._options = options |
| 3815 | variant_tensor = input_dataset._variant_tensor # pylint: disable=protected-access |
| 3816 | super(_OptionsDataset, self).__init__(input_dataset, variant_tensor) |
| 3817 | |
| 3818 | def options(self): |
| 3819 | return self._options |