(self, dataset)
| 2099 | """Wraps a V2 `Dataset` object in the `tf.compat.v1.data.Dataset` API.""" |
| 2100 | |
| 2101 | def __init__(self, dataset): |
| 2102 | self._dataset = dataset |
| 2103 | super(DatasetV1Adapter, self).__init__() |
| 2104 | |
| 2105 | def _as_variant_tensor(self): |
| 2106 | return self._dataset._variant_tensor # pylint: disable=protected-access |