fix: refresh tree return crash

dooit-org · Sep 21, 2024 · 84904a1 · 84904a1
1 parent 37b3981
commit 84904a1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dooit/ui/widgets/trees/_decorators.py b/dooit/ui/widgets/trees/_decorators.py
@@ -29,8 +29,9 @@ def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
 
 def refresh_tree(func: Callable) -> Callable:
     def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
-        func(self, *args, **kwargs)
+        res = func(self, *args, **kwargs)
         self.force_refresh()
+        return res
 
     return wrapper