diff options
author | Jiri Simsa <jsimsa@google.com> | 2018-06-18 09:10:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-18 09:12:57 -0700 |
commit | e006d39bf0021f3af2ebcf9c3c983070bf444818 (patch) | |
tree | 0c2e44d4eb12762ab286b34d530fe3786e92bb4e /tensorflow/contrib/training | |
parent | 147eb9db850dbd50dcb2ac5aa52c51396b82c4c0 (diff) |
[tf.data] Cleanup of tf.contrib.data python tests.
PiperOrigin-RevId: 201004909
Diffstat (limited to 'tensorflow/contrib/training')
-rw-r--r-- | tensorflow/contrib/training/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/contrib/training/python/training/tensor_queue_dataset_test.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD index 5de55b5f7f..76927e62e8 100644 --- a/tensorflow/contrib/training/BUILD +++ b/tensorflow/contrib/training/BUILD @@ -295,7 +295,7 @@ py_test( tags = ["notsan"], deps = [ ":training_py", - "//tensorflow/contrib/data/python/kernel_tests:dataset_serialization_test", + "//tensorflow/contrib/data/python/kernel_tests/serialization:dataset_serialization_test_base", "//tensorflow/python:client_testlib", "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:gradients", diff --git a/tensorflow/contrib/training/python/training/tensor_queue_dataset_test.py b/tensorflow/contrib/training/python/training/tensor_queue_dataset_test.py index 0338f409a2..df0a186f4f 100644 --- a/tensorflow/contrib/training/python/training/tensor_queue_dataset_test.py +++ b/tensorflow/contrib/training/python/training/tensor_queue_dataset_test.py @@ -19,7 +19,7 @@ from __future__ import print_function import numpy as np -from tensorflow.contrib.data.python.kernel_tests import dataset_serialization_test_base +from tensorflow.contrib.data.python.kernel_tests.serialization import dataset_serialization_test_base from tensorflow.contrib.training.python.training import tensor_queue_dataset as tqd from tensorflow.python.data.ops import dataset_ops from tensorflow.python.framework import dtypes |