Fix linting

deepset-ai · Jan 16, 2024 · eda4d93 · eda4d93
1 parent ceecf38
commit eda4d93
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/integrations/weaviate/tests/test_document_store.py b/integrations/weaviate/tests/test_document_store.py
@@ -1,5 +1,7 @@
-from unittest.mock import patch, MagicMock
+from unittest.mock import MagicMock, patch
 
+import pytest
+from haystack.testing.document_store import CountDocumentsTest, WriteDocumentsTest
 from haystack_integrations.document_stores.weaviate.document_store import WeaviateDocumentStore
 from weaviate.auth import AuthApiKey
 from weaviate.config import Config