fix mypy errors

basxsoftwareassociation · Dec 21, 2021 · ec65880 · ec65880
1 parent 5f10984
commit ec65880
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 6 deletions.
diff --git a/htmlgenerator/base.py b/htmlgenerator/base.py
@@ -270,7 +270,7 @@ def treewalk(
         typing.Callable[[BaseElement, typing.Tuple[BaseElement, ...]], bool]
     ],
     apply: typing.Optional[
-        typing.Callable[[BaseElement, int, BaseElement], None]
+        typing.Callable[[typing.List, int, BaseElement], None]
     ] = None,
 ) -> typing.Generator[BaseElement, None, None]:
     from .htmltags import HTMLElement

diff --git a/htmlgenerator/contrib/convertfromhtml.py b/htmlgenerator/contrib/convertfromhtml.py
@@ -1,6 +1,6 @@
 import codecs
 
-import black
+import black  # type: ignore
 from bs4 import BeautifulSoup, Comment, Doctype, NavigableString, Tag  # type: ignore
 
 # be aware: attributes with the empty string as value will be converted

diff --git a/htmlgenerator/htmltags.py b/htmlgenerator/htmltags.py
@@ -23,7 +23,9 @@ def __init__(
         super().__init__(*children)
         self.lazy_attributes = lazy_attributes
 
-    def render(self, context: dict) -> typing.Generator[str, None, None]:
+    def render(
+        self, context: dict, stringify: bool = True
+    ) -> typing.Generator[str, None, None]:
         attr_str = flatattrs(
             {
                 **self.attributes,
@@ -53,7 +55,9 @@ class VoidElement(HTMLElement):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
-    def render(self, context) -> typing.Generator[str, None, None]:
+    def render(
+        self, context, stringify: bool = True
+    ) -> typing.Generator[str, None, None]:
         attr_str = flatattrs(
             {
                 **self.attributes,
@@ -338,7 +342,9 @@ def __init__(self, *args, doctype=False, **kwargs):
         super().__init__(*args, **kwargs)
         self.doctype = doctype
 
-    def render(self, context: dict) -> typing.Generator[str, None, None]:
+    def render(
+        self, context: dict, stringify: bool = True
+    ) -> typing.Generator[str, None, None]:
         if self.doctype:
             yield "<!DOCTYPE html>"
         yield from super().render(context)

diff --git a/setup.py b/setup.py
@@ -1,4 +1,4 @@
-from setuptools import find_packages, setup
+from setuptools import find_packages, setup  # type: ignore
 
 with open("README.md") as f:
     long_description = f.read()