clean: simplify

dooit-org · Sep 16, 2024 · 2dc5753 · 2dc5753
1 parent c1b7f98
commit 2dc5753
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/dooit/ui/widgets/trees/_decorators.py b/dooit/ui/widgets/trees/_decorators.py
@@ -4,10 +4,8 @@
 if TYPE_CHECKING:
     from .model_tree import ModelTree
 
-    ModelTreeFunc = Callable[..., Any]
 
-
-def fix_highlight(func: "ModelTreeFunc") -> "ModelTreeFunc":
+def fix_highlight(func: Callable) -> Callable:
 
     def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
         highlighted_id = self.node.id
@@ -25,7 +23,7 @@ def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
     return wrapper
 
 
-def refresh_tree(func: "ModelTreeFunc") -> "ModelTreeFunc":
+def refresh_tree(func: Callable) -> Callable:
 
     def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
         func(self, *args, **kwargs)