linting

vizro-ai/src/vizro_ai/_vizro_ai.py

@@ -8,7 +8,7 @@
 
 from vizro_ai.chains._llm_models import _get_llm_model
 from vizro_ai.components import GetCodeExplanation, GetDebugger
-from vizro_ai.dashboard.graph.code_generation import _create_and_compile_graph, GraphState
+from vizro_ai.dashboard.graph.code_generation import GraphState, _create_and_compile_graph
 from vizro_ai.task_pipeline._pipeline_manager import PipelineManager
 from vizro_ai.utils.helper import (
     DebugFailure,

vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py

@@ -4,11 +4,10 @@
 import logging
 import operator
 import re
-import inspect
 from typing import Annotated, Any, Dict, List, Union
 
 import pandas as pd
-from langchain_core.messages import BaseMessage, HumanMessage, FunctionMessage
+from langchain_core.messages import BaseMessage, FunctionMessage, HumanMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.constants import END, Send
 from langgraph.graph import StateGraph
@@ -210,7 +209,7 @@ def _create_and_compile_graph():
     graph.add_conditional_edges("_dashboard_plan", continue_to_pages)
     graph.add_edge("build_page", "build_dashboard")
 
-    # temprorarily removed the node _generate_dashboard_code, will add it back once
+    # temporarily removed the node _generate_dashboard_code, will add it back once
     # the code to string is ready
     # graph.add_edge("build_dashboard", "_generate_dashboard_code")
     # graph.add_edge("_generate_dashboard_code", END)

vizro-ai/src/vizro_ai/dashboard/nodes/core_builder/model.py

@@ -6,7 +6,6 @@
 except ImportError:  # pragma: no cov
     from pydantic import BaseModel, Field, ValidationError
 from typing import Dict
-from vizro.models import VizroBaseModel
 
 from langchain_core.messages import HumanMessage
 from langchain_core.prompts import ChatPromptTemplate

vizro-ai/src/vizro_ai/dashboard/nodes/core_builder/plan.py

@@ -3,9 +3,9 @@
 import logging
 from typing import Dict, List, Literal, Union
 
+from langchain_openai import ChatOpenAI
 from vizro.models import AgGrid, Filter, Graph, Layout
 from vizro.models.types import ComponentType
-from langchain_openai import ChatOpenAI
 
 try:
     from pydantic.v1 import BaseModel, Field, ValidationError, create_model, validator