fix misc types

Author: rmorshea

idom/core/element.py

@@ -13,7 +13,7 @@
 ElementRenderFunction = Callable[..., Awaitable["VdomDict"]]
 
 
-def element(function: ElementRenderFunction) -> Callable[..., Any]:
+def element(function: ElementRenderFunction) -> Callable[..., "Element"]:
     """A decorator for defining an :class:`Element`.
 
     Parameters:

idom/core/hooks.py

@@ -96,7 +96,9 @@ def use_effect(
 
 
 @overload
-def use_effect(function: _EffectApplyFunc, args: Optional[Sequence[Any]]) -> None:
+def use_effect(
+    function: _EffectApplyFunc, args: Optional[Sequence[Any]] = None
+) -> None:
     ...
 
 
@@ -179,7 +181,7 @@ def use_callback(
 
 @overload
 def use_callback(
-    function: _CallbackFunc, args: Optional[Sequence[Any]]
+    function: _CallbackFunc, args: Optional[Sequence[Any]] = None
 ) -> _CallbackFunc:
     ...
 
@@ -224,7 +226,7 @@ def use_memo(
 
 @overload
 def use_memo(
-    function: Callable[[], _StateType], args: Optional[Sequence[Any]]
+    function: Callable[[], _StateType], args: Optional[Sequence[Any]] = None
 ) -> _StateType:
     ...
 

idom/widgets/utils.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Any, Callable, Tuple, Optional, Dict, Union, Set
+from typing import Any, Callable, Tuple, Optional, Dict, Union, Set, cast
 
 from typing_extensions import Protocol
 
@@ -183,7 +183,7 @@ async def HotSwap() -> Any:
             # new displays will adopt the latest constructor and arguments
             (f, a, kw), set_state = hooks.use_state(constructor_and_arguments.current)
 
-            def add_callback():
+            def add_callback() -> Callable[[], None]:
                 set_state_callbacks.add(set_state)
                 return lambda: set_state_callbacks.remove(set_state)
 
@@ -210,7 +210,7 @@ def swap(_func_: ElementConstructor, *args: Any, **kwargs: Any) -> None:
             constructor_and_arguments.current = (_func_, args, kwargs)
             return None
 
-    return swap, HotSwap
+    return cast(MountFunc, swap), HotSwap
 
 
 def multiview() -> Tuple["MultiViewMount", ElementConstructor]:
@@ -290,36 +290,3 @@ def _add_view(
         kwargs: Dict[str, Any],
     ) -> None:
         self._views[view_id] = lambda: constructor(*args, **kwargs)
-
-
-{
-    "tagName": "div",
-    "children": [
-        {
-            "tagName": "div",
-            "children": [
-                {
-                    "tagName": "button",
-                    "attributes": {"id": "incr-button"},
-                    "children": ["click to increment"],
-                    "eventHandlers": {
-                        "onClick": {
-                            "target": "139775953392160",
-                            "preventDefault": False,
-                            "stopPropagation": False,
-                        }
-                    },
-                },
-                {
-                    "tagName": "div",
-                    "attributes": {"id": "count-is-1"},
-                    "children": ["1"],
-                },
-            ],
-        }
-    ],
-}
-[
-    {"path": "/children/1/attributes/id", "op": "replace", "value": "count-is-1"},
-    {"path": "/children/1/children/0", "op": "replace", "value": "1"},
-]