better type hints for layouts

dooit-org · Jul 24, 2024 · 5c7cc21 · 5c7cc21
1 parent 3031baf
commit 5c7cc21
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dooit/ui/api/components.py b/dooit/ui/api/components.py
@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import List, Tuple, Callable
+from typing import List, Tuple, Callable, Union
 
 
 class WorkspaceComponent(Enum):
@@ -15,5 +15,5 @@ class TodoComponent(Enum):
     effort = "effort"
 
 
-WorkspaceLayout = List[Tuple[WorkspaceComponent, Callable]]
-TodoLayout = List[Tuple[TodoComponent, Callable]]
+WorkspaceLayout = List[Union[Tuple[WorkspaceComponent, Callable], WorkspaceComponent]]
+TodoLayout = List[Union[Tuple[TodoComponent, Callable], TodoComponent]]