diff --git a/mediapipe/calculators/tensor/BUILD b/mediapipe/calculators/tensor/BUILD index ae8a0cbf..e953342d 100644 --- a/mediapipe/calculators/tensor/BUILD +++ b/mediapipe/calculators/tensor/BUILD @@ -307,6 +307,27 @@ cc_library( alwayslink = 1, ) +cc_test( + name = "universal_sentence_encoder_preprocessor_calculator_test", + srcs = ["universal_sentence_encoder_preprocessor_calculator_test.cc"], + data = ["//mediapipe/tasks/testdata/text:universal_sentence_encoder_qa"], + deps = [ + ":universal_sentence_encoder_preprocessor_calculator", + "//mediapipe/framework:calculator_framework", + "//mediapipe/framework:packet", + "//mediapipe/framework/formats:tensor", + "//mediapipe/framework/port:gtest_main", + "//mediapipe/framework/port:parse_text_proto", + "//mediapipe/framework/port:status", + "//mediapipe/framework/tool:options_map", + "//mediapipe/tasks/cc/core:utils", + "//mediapipe/tasks/cc/metadata:metadata_extractor", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + ], +) + mediapipe_proto_library( name = "inference_calculator_proto", srcs = ["inference_calculator.proto"],