fix build targets

2021-03-04 21:53:29 +05:30 · 2021-03-04 21:53:29 +05:30 · 5ede799244
commit 5ede799244
parent cda3d38821
2 changed files with 2 additions and 0 deletions
--- a/tensorflow/python/data/experimental/kernel_tests/BUILD
+++ b/tensorflow/python/data/experimental/kernel_tests/BUILD
@ -394,6 +394,7 @@ tf_py_test(
        "//tensorflow/python:script_ops",
        "//tensorflow/python:sparse_tensor",
        "//tensorflow/python/data/experimental/ops:batching",
+        "//tensorflow/python/data/kernel_tests:checkpoint_test_base",
        "//tensorflow/python/data/kernel_tests:test_base",
        "//tensorflow/python/data/ops:dataset_ops",
        "//third_party/py/numpy",
--- a/tensorflow/python/data/kernel_tests/BUILD
+++ b/tensorflow/python/data/kernel_tests/BUILD
@ -438,6 +438,7 @@ tf_py_test(
    srcs = ["map_test.py"],
    shard_count = 4,
    deps = [
+        ":checkpoint_test_base",
        ":test_base",
        "//tensorflow/core:protos_all_py",
        "//tensorflow/python:array_ops",