Fix linting

deepset-ai · Jan 16, 2024 · 0ff56f1 · 0ff56f1
1 parent 6ab41a0
commit 0ff56f1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/integrations/weaviate/pyproject.toml b/integrations/weaviate/pyproject.toml
@@ -168,7 +168,7 @@ exclude_lines = [
 
 [[tool.mypy.overrides]]
 module = [
-  "haystack.*",
+  "haystack_integrations.*",
   "pytest.*",
   "weaviate.*",
   "numpy",

diff --git a/integrations/weaviate/src/haystack_integrations/document_stores/weaviate/document_store.py b/integrations/weaviate/src/haystack_integrations/document_stores/weaviate/document_store.py
@@ -4,10 +4,11 @@
 from dataclasses import asdict
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-import weaviate
 from haystack.core.serialization import default_from_dict, default_to_dict
 from haystack.dataclasses.document import Document
 from haystack.document_stores.protocol import DuplicatePolicy
+
+import weaviate
 from weaviate.auth import AuthCredentials
 from weaviate.config import Config, ConnectionConfig
 from weaviate.embedded import EmbeddedOptions

diff --git a/integrations/weaviate/tests/test_document_store.py b/integrations/weaviate/tests/test_document_store.py
@@ -1,5 +1,6 @@
 from unittest.mock import patch
 
+from haystack_integrations.document_stores.weaviate.document_store import WeaviateDocumentStore
 from weaviate.auth import AuthApiKey
 from weaviate.config import Config
 from weaviate.embedded import (
@@ -10,8 +11,6 @@
     EmbeddedOptions,
 )
 
-from haystack_integrations.document_stores.weaviate.document_store import WeaviateDocumentStore
-
 
 class TestWeaviateDocumentStore:
     @patch("haystack_integrations.document_stores.weaviate.document_store.weaviate")