diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2018-07-02 18:35:22 +0000 |
---|---|---|
committer | Yong Tang <yong.tang.github@outlook.com> | 2018-07-02 18:35:22 +0000 |
commit | 36db4d00e46c6ebecaf9e5a73ca1994df98086de (patch) | |
tree | fddd95649b8bd244a4ab5b07168e596a12deaa44 /tensorflow/contrib/hadoop | |
parent | 7e47b7733549b99dfa14aa0592eadbf384f0f036 (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.cc | 7 | ||||
-rw-r--r-- | tensorflow/contrib/hadoop/python/kernel_tests/hadoop_test.py | 1 |
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 |