diff --git a/vizro-ai/src/vizro_ai/_vizro_ai.py b/vizro-ai/src/vizro_ai/_vizro_ai.py index 61e2dc48b..e73a11d5d 100644 --- 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 index 48cf30946..1d24be4e8 100644 --- 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 index e6bc021b8..088cb291d 100644 --- 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 index 8713536db..6f092d3db 100644 --- 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,