fix: alert style name in widget

ydataai · Oct 15, 2024 · e4814fd · e4814fd
1 parent 5d7b7be
commit e4814fd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/ydata_profiling/report/presentation/flavours/widget/alerts.py b/src/ydata_profiling/report/presentation/flavours/widget/alerts.py
@@ -29,10 +29,15 @@ def render(self) -> widgets.GridBox:
                     )
                 )
             )
+
+            style_name = styles[type_name]
+            if style_name not in ('primary', 'success', 'info', 'warning', 'danger'):
+                style_name = ''
+
             items.append(
                 Button(
                     description=type_name.replace("_", " ").capitalize(),
-                    button_style=styles[type_name],
+                    button_style=style_name,
                     disabled=True,
                 )
             )