From 948d35f539b9ae1de06a94cfc2d027181c53f5d1 Mon Sep 17 00:00:00 2001
From: Khaleel Al-Adhami <khaleel.aladhami@gmail.com>
Date: Wed, 30 Oct 2024 15:22:14 -0700
Subject: [PATCH] get it right pyright

---
 reflex/app.py             | 2 +-
 tests/units/test_app.py   | 2 +-
 tests/units/test_state.py | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/reflex/app.py b/reflex/app.py
index fcc6c0828..62e6c73dd 100644
--- a/reflex/app.py
+++ b/reflex/app.py
@@ -270,7 +270,7 @@ class App(MiddlewareMixin, LifespanMixin):
     _state_manager: Optional[StateManager] = None
 
     # Mapping from a route to event handlers to trigger when the page loads. PRIVATE.
-    load_events: Dict[str, List[IndividualEventType[[...]]]] = dataclasses.field(
+    load_events: Dict[str, List[IndividualEventType[[]]]] = dataclasses.field(
         default_factory=dict
     )
 
diff --git a/tests/units/test_app.py b/tests/units/test_app.py
index b84955d8a..1e34a67c3 100644
--- a/tests/units/test_app.py
+++ b/tests/units/test_app.py
@@ -1211,7 +1211,7 @@ async def test_process_events(mocker, token: str):
     ],
 )
 def test_overlay_component(
-    state: State | None,
+    state: Type[State] | None,
     overlay_component: Component | ComponentCallable | None,
     exp_page_child: Type[Component] | None,
 ):
diff --git a/tests/units/test_state.py b/tests/units/test_state.py
index 8397954cf..252e26bb4 100644
--- a/tests/units/test_state.py
+++ b/tests/units/test_state.py
@@ -2724,6 +2724,7 @@ class OnLoadState(State):
 
     num: int = 0
 
+    @rx.event
     def test_handler(self):
         """Test handler."""
         self.num += 1