diff options
Diffstat (limited to 'tensorflow/python/data/kernel_tests/BUILD')
| -rw-r--r-- | tensorflow/python/data/kernel_tests/BUILD | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index 23c98247bf..631b87a718 100644 --- a/tensorflow/python/data/kernel_tests/BUILD +++ b/tensorflow/python/data/kernel_tests/BUILD @@ -137,6 +137,8 @@ tf_py_test( size = "small", srcs = ["interleave_dataset_op_test.py"], additional_deps = [ + "@absl_py//absl/testing:parameterized", + "//third_party/py/numpy", "//tensorflow/python:array_ops", "//tensorflow/python:client_testlib", "//tensorflow/python:dtypes", @@ -154,6 +156,7 @@ tf_py_test( size = "small", srcs = ["map_dataset_op_test.py"], additional_deps = [ + "@absl_py//absl/testing:parameterized", "//third_party/py/numpy", "//tensorflow/python:array_ops", "//tensorflow/python:client_testlib", |