Adds unit tests for TextEmbedder.

PiperOrigin-RevId: 486997035
2022-11-08 11:01:04 -08:00
parent ff6ca706c6
commit 7a1e55b872
2 changed files with 28 additions and 4 deletions
@@ -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"],