From 7a1e55b87245b11ef18f0aa88905e699877647ba Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Tue, 8 Nov 2022 10:59:38 -0800
Subject: [PATCH] Adds unit tests for TextEmbedder.

PiperOrigin-RevId: 486997035
---
 mediapipe/tasks/testdata/text/BUILD |  6 ++++++
 third_party/external_files.bzl      | 26 ++++++++++++++++++++++----
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/mediapipe/tasks/testdata/text/BUILD b/mediapipe/tasks/testdata/text/BUILD
index 14999a03..081e63c2 100644
--- a/mediapipe/tasks/testdata/text/BUILD
+++ b/mediapipe/tasks/testdata/text/BUILD
@@ -28,6 +28,7 @@ mediapipe_files(srcs = [
     "bert_text_classifier.tflite",
     "mobilebert_embedding_with_metadata.tflite",
     "mobilebert_with_metadata.tflite",
+    "regex_one_embedding_with_metadata.tflite",
     "test_model_text_classifier_bool_output.tflite",
     "test_model_text_classifier_with_regex_tokenizer.tflite",
     "universal_sentence_encoder_qa_with_metadata.tflite",
@@ -92,6 +93,11 @@ filegroup(
     srcs = ["mobilebert_embedding_with_metadata.tflite"],
 )
 
+filegroup(
+    name = "regex_embedding_with_metadata",
+    srcs = ["regex_one_embedding_with_metadata.tflite"],
+)
+
 filegroup(
     name = "universal_sentence_encoder_qa",
     data = ["universal_sentence_encoder_qa_with_metadata.tflite"],
diff --git a/third_party/external_files.bzl b/third_party/external_files.bzl
index de1d0a97..d5ddc8d7 100644
--- a/third_party/external_files.bzl
+++ b/third_party/external_files.bzl
@@ -402,8 +402,8 @@ def external_files():
 
     http_file(
         name = "com_google_mediapipe_labels_txt",
-        sha256 = "4b9b26392f765e7a872372131cd4cee8ad7c02e496b5a1228279619b138c4b7a",
-        urls = ["https://storage.googleapis.com/mediapipe-assets/labels.txt?generation=1667855388142641"],
+        sha256 = "536feacc519de3d418de26b2effb4d75694a8c4c0063e36499a46fa8061e2da9",
+        urls = ["https://storage.googleapis.com/mediapipe-assets/labels.txt?generation=1667888034706429"],
     )
 
     http_file(
@@ -552,8 +552,14 @@ def external_files():
 
     http_file(
         name = "com_google_mediapipe_movie_review_json",
-        sha256 = "89ad347ad1cb7c587da144de6efbadec1d3e8ff0cd13e379dd16661a8186fbb5",
-        urls = ["https://storage.googleapis.com/mediapipe-assets/movie_review.json?generation=1667855392734031"],
+        sha256 = "c09b88af05844cad5133b49744fed3a0bd514d4a1c75b9d2f23e9a40bd7bc04e",
+        urls = ["https://storage.googleapis.com/mediapipe-assets/movie_review.json?generation=1667888039053188"],
+    )
+
+    http_file(
+        name = "com_google_mediapipe_movie_review_labels_txt",
+        sha256 = "4b9b26392f765e7a872372131cd4cee8ad7c02e496b5a1228279619b138c4b7a",
+        urls = ["https://storage.googleapis.com/mediapipe-assets/movie_review_labels.txt?generation=1667888041670721"],
     )
 
     http_file(
@@ -688,6 +694,18 @@ def external_files():
         urls = ["https://storage.googleapis.com/mediapipe-assets/README.md?generation=1661875904887163"],
     )
 
+    http_file(
+        name = "com_google_mediapipe_regex_one_embedding_with_metadata_tflite",
+        sha256 = "b8f5d6d090c2c73984b2b92cd2fda27e5630562741a93d127b9a744d60505bc0",
+        urls = ["https://storage.googleapis.com/mediapipe-assets/regex_one_embedding_with_metadata.tflite?generation=1667888045310541"],
+    )
+
+    http_file(
+        name = "com_google_mediapipe_regex_vocab_txt",
+        sha256 = "b1134b10927a53ce4224bbc30ccf075c9969c94ebf40c368966d1dcf445ca923",
+        urls = ["https://storage.googleapis.com/mediapipe-assets/regex_vocab.txt?generation=1667888047885461"],
+    )
+
     http_file(
         name = "com_google_mediapipe_right_hands_jpg",
         sha256 = "240c082e80128ff1ca8a83ce645e2ba4d8bc30f0967b7991cf5fa375bab489e1",