diff --git a/integrations/weaviate/tests/test_document_store.py b/integrations/weaviate/tests/test_document_store.py index 736798067..cf9927a3d 100644 --- a/integrations/weaviate/tests/test_document_store.py +++ b/integrations/weaviate/tests/test_document_store.py @@ -2,6 +2,7 @@ # # SPDX-License-Identifier: Apache-2.0 +import base64 import os from typing import List from unittest.mock import MagicMock, patch @@ -17,7 +18,7 @@ FilterDocumentsTest, FilterDocumentsTestWithDataframe, WriteDocumentsTest, - create_filterable_docs + create_filterable_docs, ) from haystack.utils.auth import Secret from numpy import array as np_array @@ -97,7 +98,6 @@ def filterable_docs_with_dataframe(self) -> List[Document]: documents[i].meta["date"] = f"{date}Z" return documents - def assert_documents_are_equal(self, received: List[Document], expected: List[Document]): assert len(received) == len(expected) received = sorted(received, key=lambda doc: doc.id)