diff options
author | Anna R <annarev@google.com> | 2018-08-29 15:07:59 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-29 15:13:34 -0700 |
commit | 62bcfd96774f3849dd01ba08dd7e9f72d38e2fe5 (patch) | |
tree | 70b4aa3a96eb953c4fa1f2c2ae7dcbcdfb69e5cf /tensorflow/python/tools | |
parent | 82491c0930922ee6f8675e0953d6a2a1ef8b9fe5 (diff) |
Pass V1 __init__.py file list when generating API. Also, check against both TENSORFLOW_API_INIT_FILES and TENSORFLOW_API_INIT_FILES_V1 in doc_srcs_test.
PiperOrigin-RevId: 210798675
Diffstat (limited to 'tensorflow/python/tools')
-rw-r--r-- | tensorflow/python/tools/api/generator/BUILD | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/tools/api/generator/BUILD b/tensorflow/python/tools/api/generator/BUILD index 36af091163..90be2cc4f7 100644 --- a/tensorflow/python/tools/api/generator/BUILD +++ b/tensorflow/python/tools/api/generator/BUILD @@ -6,6 +6,7 @@ licenses(["notice"]) # Apache 2.0 load("//tensorflow:tensorflow.bzl", "py_test") load("//tensorflow/python/tools/api/generator:api_gen.bzl", "ESTIMATOR_API_INIT_FILES") load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "TENSORFLOW_API_INIT_FILES") +load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "TENSORFLOW_API_INIT_FILES_V1") exports_files( [ @@ -55,7 +56,7 @@ py_test( args = [ "--package=tensorflow.python", "--api_name=tensorflow", - ] + TENSORFLOW_API_INIT_FILES, + ] + TENSORFLOW_API_INIT_FILES + TENSORFLOW_API_INIT_FILES_V1, main = "doc_srcs_test.py", srcs_version = "PY2AND3", deps = [ |