diff options
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 18 |
1 files changed, 14 insertions, 4 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 6d134dbb80..8d0d9f14bc 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -398,6 +398,7 @@ config_setting( package_group( name = "internal", packages = [ + "-//third_party/tensorflow/python/estimator", "//learning/meta_rank/...", "//tensorflow/...", "//tensorflow_fold/llgtm/...", @@ -546,11 +547,20 @@ gen_api_init_files( py_library( name = "tensorflow_py", - srcs = [ - ":tensorflow_python_api_gen", - "//tensorflow/python/estimator/api:estimator_python_api_gen", + srcs = ["//tensorflow/python/estimator/api:estimator_python_api_gen"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + ":tensorflow_py_no_contrib", + "//tensorflow/contrib:contrib_py", + "//tensorflow/python/estimator:estimator_py", ], +) + +py_library( + name = "tensorflow_py_no_contrib", + srcs = [":tensorflow_python_api_gen"], srcs_version = "PY2AND3", visibility = ["//visibility:public"], - deps = ["//tensorflow/python"], + deps = ["//tensorflow/python:no_contrib"], ) |