fix: return the proper type for widgets

ydataai · Dec 21, 2022 · 4c0b358 · 4c0b358
1 parent 484e0a6
commit 4c0b358
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/pandas_profiling/profile_report.py b/src/pandas_profiling/profile_report.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 import pandas as pd
+from ipywidgets.widgets.widget_box import VBox
 from tqdm.auto import tqdm
 from typeguard import typechecked
 from visions import VisionsTypeset
@@ -23,7 +24,6 @@
 from pandas_profiling.model.typeset import ProfilingTypeSet
 from pandas_profiling.report import get_report_structure
 from pandas_profiling.report.presentation.core import Root
-from pandas_profiling.report.presentation.core.renderable import Renderable
 from pandas_profiling.report.presentation.flavours.html.templates import (
     create_html_assets,
 )
@@ -234,7 +234,7 @@ def json(self) -> str:
         return self._json
 
     @property
-    def widgets(self) -> Renderable:
+    def widgets(self) -> VBox:
         if (
             isinstance(self.description_set["table"]["n"], list)
             and len(self.description_set["table"]["n"]) > 1
@@ -357,7 +357,7 @@ def _render_html(self) -> str:
             pbar.update()
         return html
 
-    def _render_widgets(self) -> Renderable:
+    def _render_widgets(self) -> VBox:
         from pandas_profiling.report.presentation.flavours import WidgetReport
 
         report = self.report