Fix annotation for parent

insolor · May 26, 2023 · 244aba8 · 244aba8
1 parent 040f847
commit 244aba8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tkinter_layout_helpers/parent_manager.py b/tkinter_layout_helpers/parent_manager.py
@@ -1,6 +1,6 @@
 import contextlib
 import tkinter as tk
-from typing import ContextManager
+from typing import ContextManager, Union
 
 
 class DefaultRootWrapper:  # pragma: no cover
@@ -17,7 +17,7 @@ def default_root(self, value: tk.Widget):
 
 
 @contextlib.contextmanager
-def set_parent(parent: tk.Widget) -> ContextManager[tk.Widget]:
+def set_parent(parent: Union[tk.Tk, tk.Toplevel, tk.Widget]) -> ContextManager[tk.Widget]:
     old_root = _default_root_wrapper.default_root
     _default_root_wrapper.default_root = parent
     try: