From 2052ff49b8a83b58539a582f0c222f07073fa11b Mon Sep 17 00:00:00 2001 From: Somashekar B R Date: Wed, 25 Sep 2024 01:18:28 +0530 Subject: [PATCH] added # type: ignore for embeddings.fireworks --- private_gpt/components/embedding/embedding_component.py | 4 +++- pyproject.toml | 3 --- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/private_gpt/components/embedding/embedding_component.py b/private_gpt/components/embedding/embedding_component.py index 4f0ae93b..945ba1f9 100644 --- a/private_gpt/components/embedding/embedding_component.py +++ b/private_gpt/components/embedding/embedding_component.py @@ -69,7 +69,9 @@ class EmbeddingComponent: ) case "fireworks": try: - from llama_index.embeddings.fireworks import FireworksEmbedding + from llama_index.embeddings.fireworks import ( # type: ignore + FireworksEmbedding, + ) except ImportError as e: raise ImportError( "FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`" diff --git a/pyproject.toml b/pyproject.toml index bfa13705..36518452 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -203,9 +203,6 @@ exclude = ["tests"] [tool.mypy-llama-index] ignore_missing_imports = true -[tool.mypy-llama_index.embeddings.fireworks] -ignore_missing_imports = true - [tool.pytest.ini_options] asyncio_mode = "auto" testpaths = ["tests"]