Adds unit tests for TextEmbedder.
PiperOrigin-RevId: 486997035
This commit is contained in:
parent
ff6ca706c6
commit
7a1e55b872
6
mediapipe/tasks/testdata/text/BUILD
vendored
6
mediapipe/tasks/testdata/text/BUILD
vendored
|
@ -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"],
|
||||
|
|
26
third_party/external_files.bzl
vendored
26
third_party/external_files.bzl
vendored
|
@ -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",
|
||||
|
|
Loading…
Reference in New Issue
Block a user