From a6637e4470fcb9b3253f77d3db9a7df361d77368 Mon Sep 17 00:00:00 2001 From: Joe Zou Date: Tue, 8 Oct 2024 15:48:37 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 683791927 --- .../python/vision/image_classifier/image_classifier_test.py | 4 +++- .../model_maker/python/vision/image_classifier/testdata/BUILD | 2 +- .../{metadata.json => image_classifier_metadata.json} | 0 3 files changed, 4 insertions(+), 2 deletions(-) rename mediapipe/model_maker/python/vision/image_classifier/testdata/{metadata.json => image_classifier_metadata.json} (100%) diff --git a/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py b/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py index a96d08ece4..e7406b93d2 100644 --- a/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py +++ b/mediapipe/model_maker/python/vision/image_classifier/image_classifier_test.py @@ -128,7 +128,9 @@ def test_create_and_train_model( 'metadata.json') output_tflite_file = os.path.join(options.hparams.export_dir, 'model.tflite') - expected_metadata_file = test_utils.get_test_data_path('metadata.json') + expected_metadata_file = test_utils.get_test_data_path( + 'image_classifier_metadata.json' + ) self.assertTrue(os.path.exists(output_tflite_file)) self.assertGreater(os.path.getsize(output_tflite_file), 0) diff --git a/mediapipe/model_maker/python/vision/image_classifier/testdata/BUILD b/mediapipe/model_maker/python/vision/image_classifier/testdata/BUILD index 3d778836aa..720faadcf2 100644 --- a/mediapipe/model_maker/python/vision/image_classifier/testdata/BUILD +++ b/mediapipe/model_maker/python/vision/image_classifier/testdata/BUILD @@ -19,5 +19,5 @@ package( filegroup( name = "testdata", - srcs = ["metadata.json"], + srcs = ["image_classifier_metadata.json"], ) diff --git a/mediapipe/model_maker/python/vision/image_classifier/testdata/metadata.json b/mediapipe/model_maker/python/vision/image_classifier/testdata/image_classifier_metadata.json similarity index 100% rename from mediapipe/model_maker/python/vision/image_classifier/testdata/metadata.json rename to mediapipe/model_maker/python/vision/image_classifier/testdata/image_classifier_metadata.json