From 5477c43f1b230cc90a960cb3afdf0e1d852c0e5b Mon Sep 17 00:00:00 2001 From: nadijagraca Date: Tue, 20 Aug 2024 09:54:47 +0200 Subject: [PATCH] addressing pr comments --- .../src/vizro_ai/dashboard/_response_models/components.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vizro-ai/src/vizro_ai/dashboard/_response_models/components.py b/vizro-ai/src/vizro_ai/dashboard/_response_models/components.py index c26fe84f1..303df2239 100644 --- a/vizro-ai/src/vizro_ai/dashboard/_response_models/components.py +++ b/vizro-ai/src/vizro_ai/dashboard/_response_models/components.py @@ -43,10 +43,10 @@ class ComponentPlan(BaseModel): def create(self, model, all_df_metadata) -> Union[vm.Card, vm.AgGrid, vm.Figure]: """Create the component.""" - from vizro_ai import VizroAI - try: if self.component_type == "Graph": + from vizro_ai import VizroAI + vizro_ai = VizroAI(model=model) return vm.Graph(