From 68407ce2d84fc975f1be274d719dac218b822de6 Mon Sep 17 00:00:00 2001
From: Khaleel Al-Adhami <khaleel.aladhami@gmail.com>
Date: Wed, 6 Nov 2024 16:56:26 -0800
Subject: [PATCH] unbreak pyi plotly (#4320)

---
 reflex/components/plotly/plotly.pyi | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/reflex/components/plotly/plotly.pyi b/reflex/components/plotly/plotly.pyi
index ba3f323f58..c9847b67ef 100644
--- a/reflex/components/plotly/plotly.pyi
+++ b/reflex/components/plotly/plotly.pyi
@@ -73,12 +73,16 @@ class Plotly(NoSSRComponent):
         on_before_hover: Optional[EventType[[], BASE_STATE]] = None,
         on_blur: Optional[EventType[[], BASE_STATE]] = None,
         on_button_clicked: Optional[EventType[[], BASE_STATE]] = None,
-        on_click: Optional[EventType[[List[Point]], BASE_STATE]] = None,
+        on_click: Optional[
+            Union[EventType[[], BASE_STATE], EventType[[List[Point]], BASE_STATE]]
+        ] = None,
         on_context_menu: Optional[EventType[[], BASE_STATE]] = None,
         on_deselect: Optional[EventType[[], BASE_STATE]] = None,
         on_double_click: Optional[EventType[[], BASE_STATE]] = None,
         on_focus: Optional[EventType[[], BASE_STATE]] = None,
-        on_hover: Optional[EventType[[List[Point]], BASE_STATE]] = None,
+        on_hover: Optional[
+            Union[EventType[[], BASE_STATE], EventType[[List[Point]], BASE_STATE]]
+        ] = None,
         on_mount: Optional[EventType[[], BASE_STATE]] = None,
         on_mouse_down: Optional[EventType[[], BASE_STATE]] = None,
         on_mouse_enter: Optional[EventType[[], BASE_STATE]] = None,
@@ -92,11 +96,17 @@ class Plotly(NoSSRComponent):
         on_relayouting: Optional[EventType[[], BASE_STATE]] = None,
         on_restyle: Optional[EventType[[], BASE_STATE]] = None,
         on_scroll: Optional[EventType[[], BASE_STATE]] = None,
-        on_selected: Optional[EventType[[List[Point]], BASE_STATE]] = None,
-        on_selecting: Optional[EventType[[List[Point]], BASE_STATE]] = None,
+        on_selected: Optional[
+            Union[EventType[[], BASE_STATE], EventType[[List[Point]], BASE_STATE]]
+        ] = None,
+        on_selecting: Optional[
+            Union[EventType[[], BASE_STATE], EventType[[List[Point]], BASE_STATE]]
+        ] = None,
         on_transition_interrupted: Optional[EventType[[], BASE_STATE]] = None,
         on_transitioning: Optional[EventType[[], BASE_STATE]] = None,
-        on_unhover: Optional[EventType[[List[Point]], BASE_STATE]] = None,
+        on_unhover: Optional[
+            Union[EventType[[], BASE_STATE], EventType[[List[Point]], BASE_STATE]]
+        ] = None,
         on_unmount: Optional[EventType[[], BASE_STATE]] = None,
         **props,
     ) -> "Plotly":