fix mypy

Author: rmorshea

src/idom/core/component.py

@@ -39,7 +39,7 @@ class Component:
 
     def __init__(
         self,
-        function: Callable[..., Union[ComponentType, VdomDict]],
+        function: Callable[..., ComponentType | VdomDict | None],
         key: Optional[Any],
         args: Tuple[Any, ...],
         kwargs: Dict[str, Any],

src/idom/core/layout.py

@@ -32,7 +32,7 @@
 
 from ._event_proxy import _wrap_in_warning_event_proxies
 from .hooks import LifeCycleHook
-from .proto import ComponentType, EventHandlerDict, VdomJson
+from .proto import ComponentType, EventHandlerDict, VdomDict, VdomJson
 from .vdom import validate_vdom_json
 
 
@@ -204,7 +204,7 @@ def _render_component(
             # wrap the model in a fragment (i.e. tagName="") to ensure components have
             # a separate node in the model state tree. This could be removed if this
             # components are given a node in the tree some other way
-            wrapper_model = {"tagName": ""}
+            wrapper_model: VdomDict = {"tagName": ""}
             if raw_model is not None:
                 wrapper_model["children"] = [raw_model]
 

src/idom/core/proto.py

@@ -87,7 +87,7 @@ class _VdomDictOptional(TypedDict, total=False):
     children: Sequence[
         # recursive types are not allowed yet:
         # https://github.com/python/mypy/issues/731
-        Union[ComponentType, Dict[str, Any], str]
+        Union[ComponentType, Dict[str, Any], str, Any]
     ]
     attributes: VdomAttributes
     eventHandlers: EventHandlerDict  # noqa

src/idom/html.py

@@ -166,10 +166,10 @@
 
 def _(*children: Any) -> VdomDict:
     """An HTML fragment - this element will not appear in the DOM"""
-    attributes, children = coalesce_attributes_and_children(children)
+    attributes, coalesced_children = coalesce_attributes_and_children(children)
     if attributes:
         raise TypeError("Fragments cannot have attributes")
-    return {"tagName": "", "children": children}
+    return {"tagName": "", "children": coalesced_children}
 
 
 # Dcument metadata