aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/hadoop
diff options
context:
space:
mode:
authorGravatar Yong Tang <yong.tang.github@outlook.com>2018-07-02 18:35:22 +0000
committerGravatar Yong Tang <yong.tang.github@outlook.com>2018-07-02 18:35:22 +0000
commit36db4d00e46c6ebecaf9e5a73ca1994df98086de (patch)
treefddd95649b8bd244a4ab5b07168e596a12deaa44 /tensorflow/contrib/hadoop
parent7e47b7733549b99dfa14aa0592eadbf384f0f036 (diff)
Update signature change and fix pylint issue
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/hadoop')
-rw-r--r--tensorflow/contrib/hadoop/kernels/hadoop_dataset_ops.cc7
-rw-r--r--tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py1
2 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/hadoop/kernels/hadoop_dataset_ops.cc b/tensorflow/contrib/hadoop/kernels/hadoop_dataset_ops.cc
index c2ddbb858d..1afb4334f5 100644
--- a/tensorflow/contrib/hadoop/kernels/hadoop_dataset_ops.cc
+++ b/tensorflow/contrib/hadoop/kernels/hadoop_dataset_ops.cc
@@ -211,7 +211,7 @@ class SequenceFileDatasetOp : public DatasetOpKernel {
filenames_(filenames),
output_types_(output_types) {}
- std::unique_ptr<IteratorBase> MakeIterator(
+ std::unique_ptr<IteratorBase> MakeIteratorInternal(
const string& prefix) const override {
return std::unique_ptr<IteratorBase>(
new Iterator({this, strings::StrCat(prefix, "::SequenceFile")}));
@@ -227,7 +227,9 @@ class SequenceFileDatasetOp : public DatasetOpKernel {
return *shapes;
}
- string DebugString() override { return "SequenceFileDatasetOp::Dataset"; }
+ string DebugString() const override {
+ return "SequenceFileDatasetOp::Dataset";
+ }
protected:
Status AsGraphDefInternal(DatasetGraphDefBuilder* b,
@@ -328,7 +330,6 @@ class SequenceFileDatasetOp : public DatasetOpKernel {
};
DataTypeVector output_types_;
};
-
}
REGISTER_KERNEL_BUILDER(Name("SequenceFileDataset").Device(DEVICE_CPU),
diff --git a/tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py b/tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py
index 8f6f4a20e3..91ba6b0a11 100644
--- a/tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py
+++ b/tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py
@@ -24,7 +24,6 @@ from tensorflow.contrib.hadoop.python.ops import hadoop_dataset_ops
from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import errors
-from tensorflow.python.ops import array_ops
from tensorflow.python.platform import resource_loader
from tensorflow.python.platform import test