Use Type[QWidget] for btn_popup

plotlyst · Feb 23, 2022 · 61b8861 · 61b8861
1 parent cff6242
commit 61b8861
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qthandy/__init__.py b/qthandy/__init__.py
@@ -1,5 +1,5 @@
 import functools
-from typing import Optional, Union
+from typing import Optional, Union, Type
 
 from qtpy.QtCore import Qt, QObject
 from qtpy.QtGui import QCursor
@@ -118,7 +118,7 @@ def gc(obj: QObject):
     obj.deleteLater()
 
 
-def btn_popup(btn: Union[QPushButton, QToolButton], popup: QWidget, show_menu_icon: bool = False):
+def btn_popup(btn: Union[QPushButton, QToolButton], popup: Type[QWidget], show_menu_icon: bool = False):
     menu = QMenu(btn)
     action = QWidgetAction(menu)
     action.setDefaultWidget(popup)