No public description

PiperOrigin-RevId: 683791927
kp-forks · Oct 8, 2024 · a6637e4 · a6637e4
1 parent 35e57e9
commit a6637e4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
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
@@ -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
@@ -19,5 +19,5 @@ package(
 
 filegroup(
     name = "testdata",
-    srcs = ["metadata.json"],
+    srcs = ["image_classifier_metadata.json"],
 )
diff --git a/...n/image_classifier/testdata/metadata.json → ...r/testdata/image_classifier_metadata.json b/...n/image_classifier/testdata/metadata.json → ...r/testdata/image_classifier_metadata.json