From 05c565898a29494be5658738d82e7c91894e89a4 Mon Sep 17 00:00:00 2001
From: Jiuqiang Tang <jqtang@google.com>
Date: Tue, 9 May 2023 12:00:13 -0700
Subject: [PATCH] Add image_segmenter_metadata_schema and
 object_detector_metadata_schema python files to the mediapipe python wheels.

PiperOrigin-RevId: 530674961
---
 setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 202917ef2..a073128e7 100644
--- a/setup.py
+++ b/setup.py
@@ -294,7 +294,12 @@ class GenerateMetadataSchema(build_ext.build_ext):
   """Generate metadata python schema files."""
 
   def run(self):
-    for target in ['metadata_schema_py', 'schema_py']:
+    for target in [
+        'image_segmenter_metadata_schema_py',
+        'metadata_schema_py',
+        'object_detector_metadata_schema_py',
+        'schema_py',
+    ]:
       bazel_command = [
           'bazel',
           'build',