Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
lingyielia committed Jun 13, 2024
1 parent e60d488 commit 714d851
Show file tree
Hide file tree
Showing 4 changed files with 2 additions and 27 deletions.
3 changes: 2 additions & 1 deletion vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@
from typing import Any, Dict, List

import pandas as pd
from langchain.globals import set_debug

# from langchain.globals import set_debug
from langgraph.graph import END, StateGraph
from vizro.models import Dashboard
from vizro_ai.chains._llm_models import _get_llm_model
Expand Down
5 changes: 0 additions & 5 deletions vizro-ai/src/vizro_ai/dashboard/nodes/core_builder/build.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,15 +49,13 @@ def _build_components(self):
)
return components


@property
def layout(self):
"""Property to get layout."""
if self._layout is None:
self._layout = self._build_layout()
return self._layout


def _build_layout(self):
logger.info(f"Building layout: {self._page_plan}")
return self._page_plan.layout.create(model=self._model, df_metadata=self._df_metadata)
Expand All @@ -81,13 +79,10 @@ def _build_controls(self):
):
control = self._page_plan.controls.controls[i].create(
model=self._model, available_components=self.available_components, df_metadata=self._df_metadata
)
model=self._model, available_components=self.available_components, df_metadata=self._df_metadata
)
if control:
controls.append(control)


return controls

@property
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from langchain_core.prompts import ChatPromptTemplate



class ProxyVizroBaseModel(BaseModel):
"""Proxy model for VizroBaseModel."""

Expand Down
20 changes: 0 additions & 20 deletions vizro-ai/src/vizro_ai/dashboard/nodes/core_builder/vizro_ai_db.py

This file was deleted.

0 comments on commit 714d851

Please sign in to comment.