final cleanup

integrations/cohere/src/cohere_haystack/embedders/document_embedder.py

@@ -5,7 +5,7 @@
 import os
 from typing import Any, Dict, List, Optional
 
-from cohere import AsyncClient, Client, COHERE_API_URL
+from cohere import COHERE_API_URL, AsyncClient, Client
 from haystack import Document, component, default_to_dict
 
 from cohere_haystack.embedders.utils import get_async_response, get_response

integrations/cohere/src/cohere_haystack/embedders/text_embedder.py

@@ -5,7 +5,7 @@
 import os
 from typing import Any, Dict, List, Optional
 
-from cohere import AsyncClient, Client, COHERE_API_URL
+from cohere import COHERE_API_URL, AsyncClient, Client
 from haystack import component, default_to_dict
 
 from cohere_haystack.embedders.utils import get_async_response, get_response

integrations/cohere/src/cohere_haystack/generator.py

@@ -6,11 +6,9 @@
 import sys
 from typing import Any, Callable, Dict, List, Optional, cast
 
-from haystack import DeserializationError, component, default_from_dict, default_to_dict
-from haystack.lazy_imports import LazyImport
-
 from cohere import COHERE_API_URL, Client
 from cohere.responses import Generations
+from haystack import DeserializationError, component, default_from_dict, default_to_dict
 
 logger = logging.getLogger(__name__)
 

integrations/cohere/tests/test_cohere_generators.py

@@ -8,7 +8,6 @@
 
 from cohere_haystack.generator import CohereGenerator
 
-
 pytestmark = pytest.mark.generators
 
 

integrations/cohere/tests/test_document_embedder.py

@@ -2,16 +2,13 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 import os
-from unittest.mock import MagicMock
 
-import numpy as np
 import pytest
-from haystack import Document
 from cohere import COHERE_API_URL
+from haystack import Document
 
 from cohere_haystack.embedders.document_embedder import CohereDocumentEmbedder
 
-
 pytestmark = pytest.mark.embedders
 
 

integrations/cohere/tests/test_text_embedder.py

@@ -2,14 +2,12 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 import os
-from unittest.mock import MagicMock
 
 import pytest
 from cohere import COHERE_API_URL
 
 from cohere_haystack.embedders.text_embedder import CohereTextEmbedder
 
-
 pytestmark = pytest.mark.embedders