fix mypy

Paperz-org · Mar 3, 2024 · 082312b · 082312b
1 parent 93328cd
commit 082312b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/blitz/ui/components/base.py b/blitz/ui/components/base.py
@@ -3,6 +3,7 @@
 
 from nicegui import ui
 from nicegui.element import Element
+
 from blitz.ui.blitz_ui import BlitzUI, get_blitz_ui
 
 
@@ -44,7 +45,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 +82,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

diff --git a/blitz/ui/components/element/base.py b/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.