diff options
Diffstat (limited to 'tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py')
| -rw-r--r-- | tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py b/tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py index 5dfb84f28e..83af31f380 100644 --- a/tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py +++ b/tensorflow/python/data/kernel_tests/concatenate_dataset_op_test.py @@ -19,6 +19,7 @@ from __future__ import print_function import numpy as np +from tensorflow.python.data.kernel_tests import test_base from tensorflow.python.data.ops import dataset_ops from tensorflow.python.data.util import nest from tensorflow.python.framework import errors @@ -26,7 +27,7 @@ from tensorflow.python.framework import tensor_shape from tensorflow.python.platform import test -class ConcatenateDatasetTest(test.TestCase): +class ConcatenateDatasetTest(test_base.DatasetTestBase): def testConcatenateDataset(self): input_components = ( |