Bugfix for display

vertica · Oct 3, 2024 · af8b59c · af8b59c
1 parent 480f967
commit af8b59c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/verticapy/jupyter/_widgets.py b/verticapy/jupyter/_widgets.py
@@ -86,14 +86,14 @@ def display(self):
         if self.orientation == "v":
             settings = widgets.HBox(self.settings_wids)
         with self.settings_box:
-            print_message(settings)
+            print_message(settings, "display")
         with self.graph_box:
             self.graph_box.clear_output(wait=True)
-            print_message(graph)
+            print_message(graph, "display")
         if self.orientation == "v":
-            print_message(widgets.VBox([self.settings_box, self.graph_box]))
+            print_message(widgets.VBox([self.settings_box, self.graph_box]), "display")
             return
-        print_message(widgets.HBox([self.settings_box, self.graph_box]))
+        print_message(widgets.HBox([self.settings_box, self.graph_box]), "display")
 
     @staticmethod
     def _accordion(children: list, titles: list) -> widgets.Accordion: