fix optional wrapping of static call methods in pyi (#4727)

reflex-dev · Feb 3, 2025 · 15da4e1 · 15da4e1
1 parent 68547dc
commit 15da4e1
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/reflex/app.py b/reflex/app.py
@@ -151,7 +151,7 @@ def default_backend_exception_handler(exception: Exception) -> EventSpec:
             position="top-center",
             id="backend_error",
             style={"width": "500px"},
-        )  # pyright: ignore [reportReturnType]
+        )
     else:
         error_message.insert(0, "An error occurred.")
         return window_alert("\n".join(error_message))

diff --git a/reflex/components/sonner/toast.pyi b/reflex/components/sonner/toast.pyi
@@ -177,7 +177,7 @@ class ToastNamespace(ComponentNamespace):
     @staticmethod
     def __call__(
         message: Union[str, Var] = "", level: Optional[str] = None, **props
-    ) -> "Optional[EventSpec]":
+    ) -> "EventSpec":
         """Send a toast message.
 
         Args:

diff --git a/reflex/utils/pyi_generator.py b/reflex/utils/pyi_generator.py
@@ -699,6 +699,7 @@ def _generate_staticmethod_call_functiondef(
             value=_get_type_hint(
                 typing.get_type_hints(clz.__call__).get("return", None),
                 type_hint_globals,
+                is_optional=False,
             )
         ),
     )

diff --git a/tests/units/test_state.py b/tests/units/test_state.py
@@ -1623,7 +1623,7 @@ def invalid_handler(self):
                         id="backend_error",
                         position="top-center",
                         style={"width": "500px"},
-                    )  # pyright: ignore [reportCallIssue, reportArgumentType]
+                    )
                 ],
                 token="",
             )