fix for latest mypy

Author: rmorshea

src/idom/core/hooks.py

@@ -46,10 +46,31 @@
 _StateType = TypeVar("_StateType")
 
 
+@overload
+def use_state(
+    initial_value: Callable[[], _StateType],
+) -> Tuple[
+    _StateType,
+    Callable[[_StateType | Callable[[_StateType], _StateType]], None],
+]:
+    ...
+
+
+@overload
+def use_state(
+    initial_value: _StateType,
+) -> Tuple[
+    _StateType,
+    Callable[[_StateType | Callable[[_StateType], _StateType]], None],
+]:
+    ...
+
+
 def use_state(
-    initial_value: Union[_StateType, Callable[[], _StateType]],
+    initial_value: _StateType | Callable[[], _StateType],
 ) -> Tuple[
-    _StateType, Callable[[Union[_StateType, Callable[[_StateType], _StateType]]], None]
+    _StateType,
+    Callable[[_StateType | Callable[[_StateType], _StateType]], None],
 ]:
     """See the full :ref:`Use State` docs for details
 

src/idom/server/prefab.py

@@ -59,7 +59,7 @@ def run(
     logger.info(f"Using {type(server).__name__}")
 
     run_server = server.run if not daemon else server.run_in_thread
-    run_server(host, port, **(run_kwargs or {}))  # type: ignore
+    run_server(host, port, **(run_kwargs or {}))
     server.wait_until_started()
 
     return server

src/idom/server/utils.py

@@ -28,7 +28,7 @@
 _FuncParams = ParamSpec("_FuncParams")
 
 
-def threaded(function: Callable[_FuncParams, None]) -> Callable[_FuncParams, Thread]:  # type: ignore
+def threaded(function: Callable[_FuncParams, None]) -> Callable[_FuncParams, Thread]:
     @wraps(function)
     def wrapper(*args: Any, **kwargs: Any) -> Thread:
         def target() -> None:

src/idom/widgets.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from base64 import b64encode
-from typing import Any, Callable, Dict, Optional, Set, Tuple, TypeVar, Union
+from typing import Any, Callable, Dict, Optional, Set, Tuple, Union
 
 import idom
 
@@ -136,13 +136,10 @@ def swap(constructor: Callable[[], Any]) -> None:
 
 
 _Func = Callable[..., Any]
-_FuncVar = TypeVar("_FuncVar", bound=_Func)
 
 
-def _use_callable(initial_func: _FuncVar) -> Tuple[_FuncVar, Callable[[_Func], None]]:
-    state: Tuple[_FuncVar, Callable[[_Func], None]] = hooks.use_state(
-        lambda: initial_func
-    )
+def _use_callable(initial_func: _Func) -> Tuple[_Func, Callable[[_Func], None]]:
+    state = hooks.use_state(lambda: initial_func)
     return state[0], lambda new: state[1](lambda old: new)