diff --git a/unstructured/embed/vertexai.py b/unstructured/embed/vertexai.py index 5228ed4973..6dc93db637 100644 --- a/unstructured/embed/vertexai.py +++ b/unstructured/embed/vertexai.py @@ -71,8 +71,6 @@ def embed_documents(self, elements: List[Element]) -> List[Element]: def _add_embeddings_to_elements(self, elements, embeddings) -> List[Element]: assert len(elements) == len(embeddings) - elements_w_embedding = [] - for i, element in enumerate(elements): - element.embeddings = embeddings[i] - elements_w_embedding.append(element) + for element, embedding in zip(elements, embeddings): + element.embeddings = embedding return elements