From bd507b2d7b4c2915c6150f96f7865e89a66ef78c Mon Sep 17 00:00:00 2001
From: Prianka Liz Kariat <prianka.kariat@ymedialabs.com>
Date: Wed, 1 Feb 2023 19:27:05 +0530
Subject: [PATCH] Updated MPPEmbeddingHelpers to reflect type change of
 quantized embeddings

---
 .../containers/utils/sources/MPPEmbedding+Helpers.mm | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/mediapipe/tasks/ios/components/containers/utils/sources/MPPEmbedding+Helpers.mm b/mediapipe/tasks/ios/components/containers/utils/sources/MPPEmbedding+Helpers.mm
index a676242e8..faf490901 100644
--- a/mediapipe/tasks/ios/components/containers/utils/sources/MPPEmbedding+Helpers.mm
+++ b/mediapipe/tasks/ios/components/containers/utils/sources/MPPEmbedding+Helpers.mm
@@ -28,12 +28,11 @@ using EmbeddingProto = ::mediapipe::tasks::components::containers::proto::Embedd
   NSString *displayName;
 
   NSMutableArray<NSNumber *> *floatEmbedding;
-  NSData *quantizedEmbedding;
+  NSMutableArray<NSNumber *> *quantizedEmbedding;
 
   if (embeddingProto.has_float_embedding()) {
     floatEmbedding =
         [NSMutableArray arrayWithCapacity:embeddingProto.float_embedding().values_size()];
-    const auto floatEmbeddingValues = embeddingProto.float_embedding().values();
 
     for (const auto value : embeddingProto.float_embedding().values()) {
       [floatEmbedding addObject:[NSNumber numberWithFloat:value]];
@@ -41,11 +40,12 @@ using EmbeddingProto = ::mediapipe::tasks::components::containers::proto::Embedd
   }
 
   if (embeddingProto.has_quantized_embedding()) {
-    const std::string &cppQuantizedEmbedding = embeddingProto.quantized_embedding().values().data();
+    const std::string &cppQuantizedEmbedding = embeddingProto.quantized_embedding().values();
+    quantizedEmbedding = [NSMutableArray arrayWithCapacity:cppQuantizedEmbedding.length()];
 
-    const char *cppQuantizedEmbeddingCString = cppQuantizedEmbedding.c_str();
-    quantizedEmbedding = [NSData dataWithBytes:cppQuantizedEmbeddingCString
-                                        length:sizeof(cppQuantizedEmbeddingCString)];
+    for (char ch : cppQuantizedEmbedding) {
+      [quantizedEmbedding addObject:[NSNumber numberWithChar:ch]];
+    }
   }
 
   NSString *headName;