From 0ff56f15312258cf4af1b07fd427d10a98e1674d Mon Sep 17 00:00:00 2001 From: Silvano Cerza Date: Tue, 16 Jan 2024 10:48:18 +0100 Subject: [PATCH] Fix linting --- integrations/weaviate/pyproject.toml | 2 +- .../document_stores/weaviate/document_store.py | 3 ++- integrations/weaviate/tests/test_document_store.py | 3 +-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/integrations/weaviate/pyproject.toml b/integrations/weaviate/pyproject.toml index ed449f75c..04d05fa94 100644 --- 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 index 263d96aad..9317fb9de 100644 --- 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 index f43bdd7b0..3c72934b1 100644 --- 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")