Fixed a bug in embedding_result

2022-11-12 08:55:56 -08:00 · 2022-11-12 08:55:56 -08:00 · a7ed160a8e
commit a7ed160a8e
parent a3788a23bc
1 changed files with 1 additions and 1 deletions
--- 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,