[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
mckinsey · Aug 5, 2024 · f0a3329 · f0a3329
1 parent 8fa940b
commit f0a3329
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/vizro-ai/src/vizro_ai/_vizro_ai.py b/vizro-ai/src/vizro_ai/_vizro_ai.py
@@ -120,7 +120,7 @@ def _get_chart_code(self, df: pd.DataFrame, user_input: str) -> str:
         # TODO retained for some chat application integration, need deprecation handling
         return self._run_plot_tasks(df, user_input, explain=False).code
 
-    def plot(  # pylint: disable=too-many-arguments  # noqa: PLR0913
+    def plot(  # pylint: disable=too-many-arguments
         self,
         df: pd.DataFrame,
         user_input: str,

diff --git a/vizro-ai/src/vizro_ai/chains/_llm_chain.py b/vizro-ai/src/vizro_ai/chains/_llm_chain.py
@@ -35,7 +35,7 @@ def execute_chain(self, input_str: str):
 class FunctionCallChain(VizroBaseChain, ABC):
     """LLM Chain with Function Calling."""
 
-    def __init__(  # noqa: PLR0913
+    def __init__(
         self,
         llm: BaseChatModel,
         raw_prompt: str,

diff --git a/vizro-ai/tests/unit/vizro-ai/plot/components/test_visual_code.py b/vizro-ai/tests/unit/vizro-ai/plot/components/test_visual_code.py
@@ -114,7 +114,7 @@ def test_instantiation(self):
     def setup_method(self, fake_llm):
         self.get_visual_code = GetVisualCode(llm=fake_llm)
 
-    def test_pre_process(self, chart_type, input_df, df_code_1, df_schema, df_sample):  # noqa: PLR0913
+    def test_pre_process(self, chart_type, input_df, df_code_1, df_schema, df_sample):
         _, partial_vars = self.get_visual_code._pre_process(chart_type=chart_type, df_code=df_code_1, df=input_df)
         assert partial_vars == {
             "chart_type": chart_type,

diff --git a/vizro-core/tests/unit/vizro/actions/_action_loop/test_get_action_loop_components.py b/vizro-core/tests/unit/vizro/actions/_action_loop/test_get_action_loop_components.py
@@ -141,7 +141,7 @@ def test_no_components(self):
         ],
         indirect=True,
     )
-    def test_all_action_loop_components(  # pylint: disable=too-many-arguments  # noqa: PLR0913
+    def test_all_action_loop_components(  # pylint: disable=too-many-arguments
         self,
         fundamental_components,
         gateway_components,