From a7ed160a8ed410ee21a1574e6261039cb37c8a94 Mon Sep 17 00:00:00 2001 From: kinaryml Date: Sat, 12 Nov 2022 08:55:56 -0800 Subject: [PATCH] Fixed a bug in embedding_result --- .../tasks/python/components/containers/embedding_result.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mediapipe/tasks/python/components/containers/embedding_result.py b/mediapipe/tasks/python/components/containers/embedding_result.py index 63c315603..999f74535 100644 --- a/mediapipe/tasks/python/components/containers/embedding_result.py +++ b/mediapipe/tasks/python/components/containers/embedding_result.py @@ -51,7 +51,7 @@ class Embedding: bytearray(pb2_obj.quantized_embedding.values)) float_embedding = np.array(pb2_obj.float_embedding.values, dtype=float) - if not quantized_embedding: + if not pb2_obj.quantized_embedding.values: return Embedding( embedding=float_embedding, head_index=pb2_obj.head_index,