Merging in from docs-Feb

ihmeuw-msca · Aug 7, 2024 · 55ae2ef · 55ae2ef
2 parents 188392e + 6cda7bd
commit 55ae2ef
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/onemod/scheduler/scheduling_utils.py b/src/onemod/scheduler/scheduling_utils.py
@@ -93,15 +93,15 @@ class TaskRegistry:
     registry: defaultdict[str, set["Task"]] = defaultdict(set)
 
     @classmethod
-    def get(cls, function_name: str) -> list["Task"]:
+    def get(cls, function_name: str) -> list[Task]:
         return list(cls.registry[function_name])
 
     @classmethod
-    def put(cls, function_name: str, task: "Task") -> None:
+    def put(cls, function_name: str, task: Task) -> None:
         cls.registry[function_name].add(task)
 
 
-def upstream_task_callback(action: Action) -> list["Task"]:
+def upstream_task_callback(action: Action) -> list[Task]:
     """
     Given an action, we should know (based on the action name) what the relevant upstream tasks
     are.