added # type: ignore for embeddings.fireworks

zylon-ai · Sep 24, 2024 · 2052ff4 · 2052ff4
1 parent b8cb49a
commit 2052ff4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/private_gpt/components/embedding/embedding_component.py b/private_gpt/components/embedding/embedding_component.py
@@ -69,7 +69,9 @@ def __init__(self, settings: Settings) -> None:
                 )
             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
@@ -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"]