aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/kafka
diff options
context:
space:
mode:
authorGravatar Jiri Simsa <jsimsa@google.com>2018-09-25 13:42:46 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-25 13:46:54 -0700
commit348478f642216cf3cbe1eb67b875252d8e6a6418 (patch)
treec4c7afd4283506b2c413e429cab02fc547e13481 /tensorflow/contrib/kafka
parent976fb3105312bb17accebcbca2ebae906bcf99fb (diff)
[tf.data] Adding a private method for (recursively) tracking dataset inputs.
PiperOrigin-RevId: 214495925
Diffstat (limited to 'tensorflow/contrib/kafka')
-rw-r--r--tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py b/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py
index a1624614d1..7129f09e8b 100644
--- a/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py
+++ b/tensorflow/contrib/kafka/python/ops/kafka_dataset_ops.py
@@ -17,15 +17,15 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
-from tensorflow.contrib.kafka.python.ops import kafka_op_loader # pylint: disable=unused-import
from tensorflow.contrib.kafka.python.ops import gen_dataset_ops
-from tensorflow.python.data.ops.dataset_ops import Dataset
+from tensorflow.contrib.kafka.python.ops import kafka_op_loader # pylint: disable=unused-import
+from tensorflow.python.data.ops import dataset_ops
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops
from tensorflow.python.framework import tensor_shape
-class KafkaDataset(Dataset):
+class KafkaDataset(dataset_ops.DatasetSource):
"""A Kafka Dataset that consumes the message.
"""