From 47de64fc1040891c4dbda5d99a92217ffeaa1e23 Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Tue, 18 Oct 2022 08:27:31 -0700
Subject: [PATCH] Open-sources TextToTensorCalculator.

PiperOrigin-RevId: 481927522
---
 mediapipe/calculators/tensor/BUILD            | 36 +++++++++++++++++++
 .../tensor/text_to_tensor_calculator_test.cc  |  4 +--
 2 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/mediapipe/calculators/tensor/BUILD b/mediapipe/calculators/tensor/BUILD
index c52e2e28..4c6e9175 100644
--- a/mediapipe/calculators/tensor/BUILD
+++ b/mediapipe/calculators/tensor/BUILD
@@ -253,6 +253,42 @@ cc_library(
     alwayslink = 1,
 )
 
+cc_library(
+    name = "text_to_tensor_calculator",
+    srcs = ["text_to_tensor_calculator.cc"],
+    visibility = [
+        "//mediapipe/framework:mediapipe_internal",
+    ],
+    deps = [
+        "//mediapipe/framework:calculator_context",
+        "//mediapipe/framework:calculator_framework",
+        "//mediapipe/framework/api2:node",
+        "//mediapipe/framework/api2:port",
+        "//mediapipe/framework/formats:tensor",
+        "@com_google_absl//absl/status",
+        "@com_google_absl//absl/strings",
+    ],
+    alwayslink = 1,
+)
+
+cc_test(
+    name = "text_to_tensor_calculator_test",
+    srcs = ["text_to_tensor_calculator_test.cc"],
+    deps = [
+        ":text_to_tensor_calculator",
+        "//mediapipe/framework:calculator_framework",
+        "//mediapipe/framework:calculator_graph",
+        "//mediapipe/framework:packet",
+        "//mediapipe/framework/formats:tensor",
+        "//mediapipe/framework/port:gtest_main",
+        "//mediapipe/framework/port:parse_text_proto",
+        "//mediapipe/framework/tool:options_map",
+        "@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"],
diff --git a/mediapipe/calculators/tensor/text_to_tensor_calculator_test.cc b/mediapipe/calculators/tensor/text_to_tensor_calculator_test.cc
index 51c3a9a0..5c1c70aa 100644
--- a/mediapipe/calculators/tensor/text_to_tensor_calculator_test.cc
+++ b/mediapipe/calculators/tensor/text_to_tensor_calculator_test.cc
@@ -67,9 +67,7 @@ absl::StatusOr<std::string> RunTextToTensorCalculator(absl::string_view text) {
         "tensor_vec has size $0, expected 1", tensor_vec.size()));
   }
   if (tensor_vec[0].element_type() != Tensor::ElementType::kChar) {
-    return absl::InvalidArgumentError(absl::Substitute(
-        "tensor has element type $0, expected $1", tensor_vec[0].element_type(),
-        Tensor::ElementType::kChar));
+    return absl::InvalidArgumentError("Expected tensor element type kChar");
   }
   const char* buffer = tensor_vec[0].GetCpuReadView().buffer<char>();
   return std::string(buffer, text.length());