internal visibility change

PiperOrigin-RevId: 353608384 Change-Id: Ibe0cfbcb7fbceff73849931ca274b2c17834bfcd
2021-01-25 02:49:08 -08:00 · 2021-01-25 02:49:08 -08:00 · 5058eacec0
commit 5058eacec0
parent aa94c61a3e
1 changed files with 4 additions and 0 deletions
--- a/tensorflow/python/platform/BUILD
+++ b/tensorflow/python/platform/BUILD
@ -186,6 +186,10 @@ py_library(
    name = "client_testlib",
    srcs = ["test.py"],
    srcs_version = "PY3",
+    visibility = visibility + [
+        # TODO(b/178183520): remove once the test no longer depend on internal files.
+        "//learning/serving/testing/prober/model_generators/tpu:__pkg__",
+    ],
    deps = [
        ":test",
        "//tensorflow/python:client",