diff --git a/src/crewai/utilities/embedding_configurator.py b/src/crewai/utilities/embedding_configurator.py index adba70747..788547c96 100644 --- a/src/crewai/utilities/embedding_configurator.py +++ b/src/crewai/utilities/embedding_configurator.py @@ -1,6 +1,6 @@ import os import warnings -from typing import Any, Dict, Optional, cast, List, Union, Callable +from typing import Any, Callable, Dict, List, Optional, Union, cast # Initialize with None to indicate module import status CHROMADB_AVAILABLE = False @@ -21,7 +21,9 @@ try: from chromadb.api.types import Documents as ChromaDocuments from chromadb.api.types import EmbeddingFunction as ChromaEmbeddingFunction from chromadb.api.types import Embeddings as ChromaEmbeddings - from chromadb.utils import validate_embedding_function as chroma_validate_embedding_function + from chromadb.utils import ( + validate_embedding_function as chroma_validate_embedding_function, + ) # Override our placeholder types with the real ones Documents = ChromaDocuments diff --git a/tests/test_numpy_compatibility.py b/tests/test_numpy_compatibility.py index 049f75176..7b1a21eb3 100644 --- a/tests/test_numpy_compatibility.py +++ b/tests/test_numpy_compatibility.py @@ -1,8 +1,10 @@ import importlib -import pytest import sys import warnings +import pytest + + def test_crew_import_with_numpy(): """Test that crewai can be imported even with NumPy compatibility issues.""" try: