Fix annotation of Heading

srittau · Aug 21, 2018 · 59e843b · 59e843b
1 parent a69cf15
commit 59e843b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/NEWS.rst b/NEWS.rst
@@ -6,6 +6,7 @@ API Additions
 
 * Make ``ElementBase`` public.
 * Add ``GenValue`` and ``GenValueGenerator`` type aliases.
+* Fix annotation of ``Heading``.
 
 News in version 1.1.0
 =====================

diff --git a/htmlgen/structure.pyi b/htmlgen/structure.pyi
@@ -1,6 +1,5 @@
-from typing import Union,  Generator
-
-from htmlgen.element import Element
+from .element import Element
+from .generator import GenValue
 
 
 class Section(Element):
@@ -22,4 +21,4 @@ class Footer(Element):
     def __init__(self) -> None: ...
 
 class Heading(Element):
-    def __init__(self, level: int = ..., *content: Union[str, bytes, Generator]) -> None: ...
+    def __init__(self, level: int = ..., *content: GenValue) -> None: ...