From 185a57cec51886cd18747273f7fcefe59636a44c Mon Sep 17 00:00:00 2001
From: Mark McDonald <macd@google.com>
Date: Thu, 10 Nov 2022 23:11:15 -0800
Subject: [PATCH] Support nested `mediapipe` dir in Java API doc builder.

PiperOrigin-RevId: 487735796
---
 docs/build_java_api_docs.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/docs/build_java_api_docs.py b/docs/build_java_api_docs.py
index 7b44e79f4..e96e1fd83 100644
--- a/docs/build_java_api_docs.py
+++ b/docs/build_java_api_docs.py
@@ -45,6 +45,12 @@ def main(_) -> None:
     while (mp_root := mp_root.parent).name != 'mediapipe':
       # Find the nearest `mediapipe` dir.
       pass
+
+    # Externally, parts of the repo are nested inside a mediapipe/ directory
+    # that does not exist internally. Support both.
+    if (mp_root / 'mediapipe').exists():
+      mp_root = mp_root / 'mediapipe'
+
     java_root = mp_root / 'tasks/java'
 
   gen_java.gen_java_docs(