From 01857d6875704ef5ce16faf0d1a2e56de2d6a5b8 Mon Sep 17 00:00:00 2001
From: mohammadrezapourreza <m1378.prz@gmail.com>
Date: Fri, 10 May 2024 10:01:57 -0400
Subject: [PATCH] DH-5776/ignore linter

---
 dataherald/services/sql_generations.py          | 4 ++--
 dataherald/sql_generator/__init__.py            | 4 ++--
 dataherald/sql_generator/dataherald_sqlagent.py | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/dataherald/services/sql_generations.py b/dataherald/services/sql_generations.py
index b1890443..2d9a8121 100644
--- a/dataherald/services/sql_generations.py
+++ b/dataherald/services/sql_generations.py
@@ -63,9 +63,9 @@ def update_the_initial_sql_generation(
         initial_sql_generation.intermediate_steps = sql_generation.intermediate_steps
         return self.sql_generation_repository.update(initial_sql_generation)
 
-    def create(
+    def create( # noqa: PLR0912
         self, prompt_id: str, sql_generation_request: SQLGenerationRequest
-    ) -> SQLGeneration:
+    ) -> SQLGeneration: # noqa: PLR0912
         initial_sql_generation = SQLGeneration(
             prompt_id=prompt_id,
             created_at=datetime.now(),
diff --git a/dataherald/sql_generator/__init__.py b/dataherald/sql_generator/__init__.py
index e997920e..94ae14ad 100644
--- a/dataherald/sql_generator/__init__.py
+++ b/dataherald/sql_generator/__init__.py
@@ -179,7 +179,7 @@ def generate_response(
         """Generates a response to a user question."""
         pass
 
-    def stream_agent_steps(  # noqa: C901
+    def stream_agent_steps(  # noqa: C901 # noqa: PLR0912
         self,
         question: str,
         agent_executor: AgentExecutor,
@@ -187,7 +187,7 @@ def stream_agent_steps(  # noqa: C901
         sql_generation_repository: SQLGenerationRepository,
         queue: Queue,
         metadata: dict = None,
-    ):
+    ): # noqa: PLR0912
         try:
             with get_openai_callback() as cb:
                 for chunk in agent_executor.stream(
diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py
index a0f490ff..6d1575f4 100644
--- a/dataherald/sql_generator/dataherald_sqlagent.py
+++ b/dataherald/sql_generator/dataherald_sqlagent.py
@@ -710,13 +710,13 @@ def create_sql_agent(
         )
 
     @override
-    def generate_response(
+    def generate_response( # noqa: PLR0912
         self,
         user_prompt: Prompt,
         database_connection: DatabaseConnection,
         context: List[dict] = None,
         metadata: dict = None,
-    ) -> SQLGeneration:
+    ) -> SQLGeneration: # noqa: PLR0912
         context_store = self.system.instance(ContextStore)
         storage = self.system.instance(DB)
         response = SQLGeneration(