From cce1751dbf7d6cd52cc080b312f080f7972150d6 Mon Sep 17 00:00:00 2001
From: Jiuqiang Tang <jqtang@google.com>
Date: Mon, 14 Nov 2022 15:39:16 -0800
Subject: [PATCH] Expose tasks components API in PyPI packages.

PiperOrigin-RevId: 488487666
---
 mediapipe/tasks/python/audio/__init__.py      |  6 +++
 .../python/components/containers/__init__.py  | 40 +++++++++++++++++++
 .../python/components/processors/__init__.py  | 13 ++++++
 mediapipe/tasks/python/text/__init__.py       |  6 +++
 4 files changed, 65 insertions(+)

diff --git a/mediapipe/tasks/python/audio/__init__.py b/mediapipe/tasks/python/audio/__init__.py
index 947f95d9d..e129800a3 100644
--- a/mediapipe/tasks/python/audio/__init__.py
+++ b/mediapipe/tasks/python/audio/__init__.py
@@ -16,12 +16,18 @@
 
 import mediapipe.tasks.python.audio.core
 import mediapipe.tasks.python.audio.audio_classifier
+import mediapipe.tasks.python.audio.audio_embedder
 
 AudioClassifier = audio_classifier.AudioClassifier
 AudioClassifierOptions = audio_classifier.AudioClassifierOptions
+AudioClassifierResult = audio_classifier.AudioClassifierResult
+AudioEmbedder = audio_embedder.AudioEmbedder
+AudioEmbedderOptions = audio_embedder.AudioEmbedderOptions
+AudioEmbedderResult = audio_embedder.AudioEmbedderResult
 RunningMode = core.audio_task_running_mode.AudioTaskRunningMode
 
 # Remove unnecessary modules to avoid duplication in API docs.
 del audio_classifier
+del audio_embedder
 del core
 del mediapipe
diff --git a/mediapipe/tasks/python/components/containers/__init__.py b/mediapipe/tasks/python/components/containers/__init__.py
index 65c1214af..17464db36 100644
--- a/mediapipe/tasks/python/components/containers/__init__.py
+++ b/mediapipe/tasks/python/components/containers/__init__.py
@@ -11,3 +11,43 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+"""MediaPipe Tasks Components Containers API."""
+
+import mediapipe.tasks.python.components.containers.audio_data
+import mediapipe.tasks.python.components.containers.bounding_box
+import mediapipe.tasks.python.components.containers.category
+import mediapipe.tasks.python.components.containers.classification_result
+import mediapipe.tasks.python.components.containers.detections
+import mediapipe.tasks.python.components.containers.embedding_result
+import mediapipe.tasks.python.components.containers.landmark
+import mediapipe.tasks.python.components.containers.landmark_detection_result
+import mediapipe.tasks.python.components.containers.rect
+
+AudioDataFormat = audio_data.AudioDataFormat
+AudioData = audio_data.AudioData
+BoundingBox = bounding_box.BoundingBox
+Category = category.Category
+Classifications = classification_result.Classifications
+ClassificationResult = classification_result.ClassificationResult
+Detection = detections.Detection
+DetectionResult = detections.DetectionResult
+Embedding = embedding_result.Embedding
+EmbeddingResult = embedding_result.EmbeddingResult
+Landmark = landmark.Landmark
+NormalizedLandmark = landmark.NormalizedLandmark
+LandmarksDetectionResult = landmark_detection_result.LandmarksDetectionResult
+Rect = rect.Rect
+NormalizedRect = rect.NormalizedRect
+
+# Remove unnecessary modules to avoid duplication in API docs.
+del audio_data
+del bounding_box
+del category
+del classification_result
+del detections
+del embedding_result
+del landmark
+del landmark_detection_result
+del rect
+del mediapipe
diff --git a/mediapipe/tasks/python/components/processors/__init__.py b/mediapipe/tasks/python/components/processors/__init__.py
index 65c1214af..adcb38757 100644
--- a/mediapipe/tasks/python/components/processors/__init__.py
+++ b/mediapipe/tasks/python/components/processors/__init__.py
@@ -11,3 +11,16 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+"""MediaPipe Tasks Components Processors API."""
+
+import mediapipe.tasks.python.components.processors.classifier_options
+import mediapipe.tasks.python.components.processors.embedder_options
+
+ClassifierOptions = classifier_options.ClassifierOptions
+EmbedderOptions = embedder_options.EmbedderOptions
+
+# Remove unnecessary modules to avoid duplication in API docs.
+del classifier_options
+del embedder_options
+del mediapipe
diff --git a/mediapipe/tasks/python/text/__init__.py b/mediapipe/tasks/python/text/__init__.py
index e2473f56b..ecf3a0ad2 100644
--- a/mediapipe/tasks/python/text/__init__.py
+++ b/mediapipe/tasks/python/text/__init__.py
@@ -15,10 +15,16 @@
 """MediaPipe Tasks Text API."""
 
 import mediapipe.tasks.python.text.text_classifier
+import mediapipe.tasks.python.text.text_embedder
 
 TextClassifier = text_classifier.TextClassifier
 TextClassifierOptions = text_classifier.TextClassifierOptions
+TextClassifierResult = text_classifier.TextClassifierResult
+TextEmbedder = text_embedder.TextEmbedder
+TextEmbedderOptions = text_embedder.TextEmbedderOptions
+TextEmbedderResult = text_embedder.TextEmbedderResult
 
 # Remove unnecessary modules to avoid duplication in API docs.
 del mediapipe
 del text_classifier
+del text_embedder