diff --git a/reflex/components/plotly/plotly.pyi b/reflex/components/plotly/plotly.pyi index fc9b2857f..4c87d589f 100644 --- a/reflex/components/plotly/plotly.pyi +++ b/reflex/components/plotly/plotly.pyi @@ -45,20 +45,20 @@ class Plotly(NoSSRComponent): class_name: Optional[Any] = None, autofocus: Optional[bool] = None, custom_attrs: Optional[Dict[str, Union[Var, str]]] = None, - on_after_plot: Optional[EventType[[Any], BASE_STATE]] = None, - on_animated: Optional[EventType[[Any], BASE_STATE]] = None, - on_animating_frame: Optional[EventType[[Any], BASE_STATE]] = None, - on_animation_interrupted: Optional[EventType[[Any], BASE_STATE]] = None, - on_autosize: Optional[EventType[[Any], BASE_STATE]] = None, - on_before_hover: Optional[EventType[[Any], BASE_STATE]] = None, + on_after_plot: Optional[EventType[..., BASE_STATE]] = None, + on_animated: Optional[EventType[..., BASE_STATE]] = None, + on_animating_frame: Optional[EventType[..., BASE_STATE]] = None, + on_animation_interrupted: Optional[EventType[..., BASE_STATE]] = None, + on_autosize: Optional[EventType[..., BASE_STATE]] = None, + on_before_hover: Optional[EventType[..., BASE_STATE]] = None, on_blur: Optional[EventType[[], BASE_STATE]] = None, - on_button_clicked: Optional[EventType[[Any], BASE_STATE]] = None, - on_click: Optional[EventType[[Any], BASE_STATE]] = None, + on_button_clicked: Optional[EventType[..., BASE_STATE]] = None, + on_click: Optional[EventType[..., BASE_STATE]] = None, on_context_menu: Optional[EventType[[], BASE_STATE]] = None, - on_deselect: Optional[EventType[[Any], BASE_STATE]] = None, - on_double_click: Optional[EventType[[Any], 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[[Any], BASE_STATE]] = None, + on_hover: Optional[EventType[..., 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, @@ -67,16 +67,16 @@ class Plotly(NoSSRComponent): on_mouse_out: Optional[EventType[[], BASE_STATE]] = None, on_mouse_over: Optional[EventType[[], BASE_STATE]] = None, on_mouse_up: Optional[EventType[[], BASE_STATE]] = None, - on_redraw: Optional[EventType[[Any], BASE_STATE]] = None, - on_relayout: Optional[EventType[[Any], BASE_STATE]] = None, - on_relayouting: Optional[EventType[[Any], BASE_STATE]] = None, - on_restyle: Optional[EventType[[Any], BASE_STATE]] = None, + on_redraw: Optional[EventType[..., BASE_STATE]] = None, + on_relayout: Optional[EventType[..., BASE_STATE]] = None, + 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[[Any], BASE_STATE]] = None, - on_selecting: Optional[EventType[[Any], BASE_STATE]] = None, - on_transition_interrupted: Optional[EventType[[Any], BASE_STATE]] = None, - on_transitioning: Optional[EventType[[Any], BASE_STATE]] = None, - on_unhover: Optional[EventType[[Any], BASE_STATE]] = None, + on_selected: Optional[EventType[..., BASE_STATE]] = None, + on_selecting: Optional[EventType[..., BASE_STATE]] = None, + on_transition_interrupted: Optional[EventType[..., BASE_STATE]] = None, + on_transitioning: Optional[EventType[..., BASE_STATE]] = None, + on_unhover: Optional[EventType[..., BASE_STATE]] = None, on_unmount: Optional[EventType[[], BASE_STATE]] = None, **props, ) -> "Plotly": diff --git a/reflex/utils/pyi_generator.py b/reflex/utils/pyi_generator.py index 32a7b65cb..d200519b7 100644 --- a/reflex/utils/pyi_generator.py +++ b/reflex/utils/pyi_generator.py @@ -496,7 +496,7 @@ def _generate_component_create_functiondef( def figure_out_return_type(annotation: Any): if inspect.isclass(annotation) and issubclass(annotation, inspect._empty): - return ast.Name(id="EventType[[Any], BASE_STATE]") + return ast.Name(id="EventType[..., BASE_STATE]") if not isinstance(annotation, str) and get_origin(annotation) is tuple: arguments = get_args(annotation) @@ -548,7 +548,7 @@ def _generate_component_create_functiondef( return ast.Name( id=f"EventType[[{', '.join(arguments_without_var)}], BASE_STATE]" ) - return ast.Name(id="EventType[[Any], BASE_STATE]") + return ast.Name(id="EventType[..., BASE_STATE]") event_triggers = clz().get_event_triggers()