aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/estimator/api/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/estimator/api/BUILD')
-rw-r--r--tensorflow/python/estimator/api/BUILD1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/api/BUILD b/tensorflow/python/estimator/api/BUILD
index aa5a29e6dd..ceb9baef4d 100644
--- a/tensorflow/python/estimator/api/BUILD
+++ b/tensorflow/python/estimator/api/BUILD
@@ -13,6 +13,7 @@ gen_api_init_files(
name = "estimator_python_api_gen",
api_name = "estimator",
output_files = ESTIMATOR_API_INIT_FILES,
+ output_package = "tensorflow.python.estimator.api",
package = "tensorflow.python.estimator",
package_dep = "//tensorflow/python/estimator:estimator_py",
)