diff options
Diffstat (limited to 'tensorflow/contrib/data/python/ops/unique.py')
| -rw-r--r-- | tensorflow/contrib/data/python/ops/unique.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/data/python/ops/unique.py b/tensorflow/contrib/data/python/ops/unique.py index e0d606311c..bad67a580d 100644 --- a/tensorflow/contrib/data/python/ops/unique.py +++ b/tensorflow/contrib/data/python/ops/unique.py @@ -47,12 +47,12 @@ def unique(): return _apply_fn -class _UniqueDataset(dataset_ops.Dataset): +class _UniqueDataset(dataset_ops.UnaryDataset): """A `Dataset` contains the unique elements from its input.""" def __init__(self, input_dataset): """See `unique()` for details.""" - super(_UniqueDataset, self).__init__() + super(_UniqueDataset, self).__init__(input_dataset) self._input_dataset = input_dataset if input_dataset.output_types not in (dtypes.int32, dtypes.int64, dtypes.string): |