Add TensorFlow __init__.py back to py_binary targets

Change: 143837911
JoeyBoyi · Jan 7, 2017 · e7ce847 · e7ce847
1 parent 1095b9b
commit e7ce847
Show file tree

Hide file tree

Showing 7 changed files with 27 additions and 11 deletions.
diff --git a/tensorflow/contrib/learn/BUILD b/tensorflow/contrib/learn/BUILD
@@ -1037,12 +1037,12 @@ py_test(
 
 py_binary(
     name = "inspect_checkpoint",
-    srcs = [
-        "python/learn/utils/inspect_checkpoint.py",
-    ],
+    srcs = ["python/learn/utils/inspect_checkpoint.py"],
     srcs_version = "PY2AND3",
     deps = [
-        "//tensorflow:tensorflow_py",
+        "//tensorflow/contrib/framework:framework_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/python:platform",
     ],
 )
 

diff --git a/tensorflow/contrib/learn/python/learn/datasets/BUILD b/tensorflow/contrib/learn/python/learn/datasets/BUILD
@@ -33,13 +33,12 @@ py_library(
 
 py_binary(
     name = "produce_small_datasets",
-    srcs = [
-        "produce_small_datasets.py",
-    ],
+    srcs = ["produce_small_datasets.py"],
     srcs_version = "PY2AND3",
     deps = [
         ":datasets",
-        "//tensorflow:tensorflow_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/python:platform",
     ],
 )
 

diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
@@ -30,7 +30,16 @@ py_library(
     name = "python",
     srcs = ["__init__.py"],
     srcs_version = "PY2AND3",
-    visibility = ["//tensorflow:__pkg__"],
+    visibility = [
+        "//tensorflow:__pkg__",
+        "//tensorflow/compiler/aot/tests:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/contrib/learn:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/contrib/learn/python/learn/datasets:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/python/debug:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/python/tools:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/tensorboard/scripts:__pkg__",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/tools/quantization:__pkg__",  # TODO(b/34059704): remove when fixed
+    ],
     deps = [
         ":array_ops",
         ":check_ops",

diff --git a/tensorflow/python/debug/BUILD b/tensorflow/python/debug/BUILD
@@ -218,6 +218,7 @@ py_binary(
     deps = [
         ":analyzer_cli",
         ":debug_data",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:platform",
     ],
 )

diff --git a/tensorflow/python/tools/BUILD b/tensorflow/python/tools/BUILD
@@ -21,8 +21,8 @@ py_binary(
     srcs = ["freeze_graph.py"],
     srcs_version = "PY2AND3",
     deps = [
-        "//tensorflow:tensorflow_py",
         "//tensorflow/core:protos_all_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:client",
         "//tensorflow/python:framework",
         "//tensorflow/python:platform",
@@ -55,6 +55,7 @@ py_binary(
     srcs = ["inspect_checkpoint.py"],
     srcs_version = "PY2AND3",
     deps = [
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:platform",
         "//tensorflow/python:pywrap_tensorflow",
     ],
@@ -123,6 +124,7 @@ py_binary(
     deps = [
         ":optimize_for_inference_lib",
         "//tensorflow/core:protos_all_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:framework",
         "//tensorflow/python:framework_for_generated_wrappers",
         "//tensorflow/python:platform",
@@ -155,7 +157,10 @@ py_binary(
     srcs = ["print_selective_registration_header.py"],
     srcs_version = "PY2AND3",
     visibility = ["//visibility:public"],
-    deps = ["//tensorflow/python:platform"],
+    deps = [
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
+        "//tensorflow/python:platform",
+    ],
 )
 
 py_test(

diff --git a/tensorflow/tensorboard/scripts/BUILD b/tensorflow/tensorboard/scripts/BUILD
@@ -25,6 +25,7 @@ py_binary(
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/core:protos_all_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client",
         "//tensorflow/python:logging_ops",

diff --git a/tensorflow/tools/quantization/BUILD b/tensorflow/tools/quantization/BUILD
@@ -20,6 +20,7 @@ py_binary(
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/core:protos_all_py",
+        "//tensorflow/python",  # TODO(b/34059704): remove when fixed
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client",
         "//tensorflow/python:framework",