From 5058eacec08e41764b0e1a5807292dc21452e063 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Mon, 25 Jan 2021 02:49:08 -0800
Subject: [PATCH] internal visibility change

PiperOrigin-RevId: 353608384
Change-Id: Ibe0cfbcb7fbceff73849931ca274b2c17834bfcd
---
 tensorflow/python/platform/BUILD | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/python/platform/BUILD b/tensorflow/python/platform/BUILD
index 108f7355679..97c88a1343c 100644
--- 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",