Let the build_pip_package tool depends on the simple console on windows.

PiperOrigin-RevId: 338710296
tensorflow · Oct 23, 2020 · 3faaca9 · 3faaca9
1 parent 5ea9f06
commit 3faaca9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tensorflow_lite_support/tools/pip_package/BUILD b/tensorflow_lite_support/tools/pip_package/BUILD
@@ -34,7 +34,14 @@ filegroup(
 sh_binary(
     name = "build_pip_package",
     srcs = ["build_pip_package.sh"],
-    data = COMMON_PIP_DEPS,
+    data = COMMON_PIP_DEPS +
+           select({
+               "@org_tensorflow//tensorflow:windows": [
+                   ":simple_console_for_windows",
+               ],
+               "//conditions:default": [
+               ],
+           }),
 )
 
 # On Windows, python binary is a zip file of runfiles tree.