diff --git a/htmlgenerator/base.py b/htmlgenerator/base.py
index 033150b..f9d9106 100644
--- 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
index ca2c916..874c888 100644
--- 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
index e641b2f..847a3ef 100644
--- 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 ""
yield from super().render(context)
diff --git a/setup.py b/setup.py
index 12b74e3..2241928 100644
--- 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()