addressing pr comments

mckinsey · Aug 20, 2024 · 5477c43 · 5477c43
1 parent 98db7ed
commit 5477c43
Showing 1 changed file with 2 additions and 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
@@ -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(