diff options
Diffstat (limited to 'tensorflow/contrib/bigtable')
-rw-r--r-- | tensorflow/contrib/bigtable/python/ops/bigtable_api.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/bigtable/python/ops/bigtable_api.py b/tensorflow/contrib/bigtable/python/ops/bigtable_api.py index 3e1b622867..cf56822ff4 100644 --- a/tensorflow/contrib/bigtable/python/ops/bigtable_api.py +++ b/tensorflow/contrib/bigtable/python/ops/bigtable_api.py @@ -575,7 +575,7 @@ def _normalize_columns(columns, provided_kwargs): return normalized -class _BigtableKeyDataset(dataset_ops.Dataset): +class _BigtableKeyDataset(dataset_ops.DatasetSource): """_BigtableKeyDataset is an abstract class representing the keys of a table. """ @@ -645,7 +645,7 @@ class _BigtableSampleKeysDataset(_BigtableKeyDataset): table=self._table._resource) # pylint: disable=protected-access -class _BigtableLookupDataset(dataset_ops.Dataset): +class _BigtableLookupDataset(dataset_ops.DatasetSource): """_BigtableLookupDataset represents a dataset that retrieves values for keys. """ @@ -678,7 +678,7 @@ class _BigtableLookupDataset(dataset_ops.Dataset): columns=self._columns) -class _BigtableScanDataset(dataset_ops.Dataset): +class _BigtableScanDataset(dataset_ops.DatasetSource): """_BigtableScanDataset represents a dataset that retrieves keys and values. """ @@ -715,7 +715,7 @@ class _BigtableScanDataset(dataset_ops.Dataset): probability=self._probability) -class _BigtableSampleKeyPairsDataset(dataset_ops.Dataset): +class _BigtableSampleKeyPairsDataset(dataset_ops.DatasetSource): """_BigtableSampleKeyPairsDataset returns key pairs from a Bigtable table. """ |