From e391c76433cf6e1b27365cde20983fe2e8457db8 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Tue, 9 May 2023 18:42:28 -0700 Subject: [PATCH] Include object_detector_metadata_schema.fbs and image_segmenter_metadata_schema.fbs PiperOrigin-RevId: 530769920 --- setup.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index a073128e7..892c6dca7 100644 --- a/setup.py +++ b/setup.py @@ -313,7 +313,11 @@ class GenerateMetadataSchema(build_ext.build_ext): _copy_to_build_lib_dir( self.build_lib, 'mediapipe/tasks/metadata/' + target + '_generated.py') - schema_file = 'mediapipe/tasks/metadata/metadata_schema.fbs' + for schema_file in [ + 'mediapipe/tasks/metadata/metadata_schema.fbs', + 'mediapipe/tasks/metadata/object_detector_metadata_schema.fbs', + 'mediapipe/tasks/metadata/image_segmenter_metadata_schema.fbs', + ]: shutil.copyfile(schema_file, os.path.join(self.build_lib + '/', schema_file))