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