Merge pull request #392 from aitomatic/lint

update Ruff linting configs
aitomatic · Nov 25, 2024 · e3c8baa · e3c8baa
2 parents 2d33cfd + b6dd37e
commit e3c8baa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/openssa/deprecated/ooda_rag/rag_ooda.py b/openssa/deprecated/ooda_rag/rag_ooda.py
@@ -1,8 +1,8 @@
 from typing import Union, ClassVar
 from uuid import uuid4
 
-from llama_index.core.retrievers import BaseRetriever  # noqa: TCH002
-from llama_index.core.schema import NodeWithScore  # noqa: TCH002
+from llama_index.core.retrievers import BaseRetriever  # noqa: TC002
+from llama_index.core.schema import NodeWithScore  # noqa: TC002
 
 from openssa.deprecated.ooda_rag.builtin_agents import (
     ContextValidator,

diff --git a/openssa/deprecated/utils/deprecated/llama_index_api.py b/openssa/deprecated/utils/deprecated/llama_index_api.py
@@ -5,7 +5,7 @@
 
 from llama_index.core import ServiceContext
 from llama_index.core.llms import OpenAI as LlamaIndexOpenAI
-from llama_index.core.llms.llm import LLM  # noqa: TCH002
+from llama_index.core.llms.llm import LLM  # noqa: TC002
 from llama_index.core.llms.openai_utils import ALL_AVAILABLE_MODELS, CHAT_MODELS
 from llama_index.embeddings.azure_openai import AzureOpenAIEmbedding
 from llama_index.embeddings.openai import OpenAIEmbedding