Add support for generating Play Services docs.
The code is internal, so the path is provided by an optional flag.
PiperOrigin-RevId: 457387974
diff --git a/tensorflow/lite/g3doc/tools/build_java_api_docs.py b/tensorflow/lite/g3doc/tools/build_java_api_docs.py
index a5e8187..c468eb7 100644
--- a/tensorflow/lite/g3doc/tools/build_java_api_docs.py
+++ b/tensorflow/lite/g3doc/tools/build_java_api_docs.py
@@ -60,6 +60,7 @@
SECTION_LABELS = {
'org.tensorflow.lite': 'Core',
'org.tensorflow.lite.support': 'Support Library',
+ 'org.tensorflow.lite.task.gms': 'Task Library (Play Services)',
'org.tensorflow.lite.task': 'Task Library',
# If we ever need other ODML packages, drop the `.image` here.
'com.google.android.odml.image': 'ODML',
@@ -115,6 +116,10 @@
overlay(resolve_nested_dir(SOURCE_PATH_METADATA, root), merged_temp_dir)
overlay(resolve_nested_dir(SOURCE_PATH_ODML, root), merged_temp_dir)
+ if gms_path := os.getenv('GMS_PATH'):
+ # Play Services code needs to be massaged into a Java directory structure.
+ overlay(gms_path, merged_temp_dir / 'org/tensorflow/lite/task/gms')
+
gen_java.gen_java_docs(
package=['org.tensorflow.lite', 'com.google.android.odml'],
source_path=merged_temp_dir,
