fix mypy

blitz/ui/components/base.py

@@ -2,7 +2,7 @@
 from typing import Any, Generic, Protocol, Self, TypeVar, cast, overload
 
 from nicegui import ui
-from nicegui.element import Element
+
 from blitz.ui.blitz_ui import BlitzUI, get_blitz_ui
 
 
@@ -44,7 +44,7 @@ def __init__(
         render: bool | None = None,
         **kwargs: Any,
     ) -> None:
-        self._ng: Element
+        self._ng: V
         self.props = props
         self.classes = classes
         self.blitz_ui: BlitzUI
@@ -81,11 +81,11 @@ def refresh(self, *args: Any, **kwargs: Any) -> None:
             self.render.refresh(*args, **kwargs)  # type: ignore
 
     @property
-    def ng(self) -> Element:
+    def ng(self) -> V:
         return self._ng
 
     @ng.setter
-    def ng(self, value: Element) -> None:
+    def ng(self, value: V) -> None:
         self._ng = value
 
     @classmethod

blitz/ui/components/element/base.py

@@ -15,7 +15,7 @@ def render(self) -> None:
         self.ng = ui.element(tag=self.tag, _client=self._client).props(self.props).classes(self.classes)
 
 
-class IFrame(BaseElement):  # type: ignore
+class IFrame(BaseElement):
     """
     IFrame element.