chore: make TaskClient.create_task return created task

ICIJ · Dec 5, 2024 · c1b84ef · c1b84ef
1 parent 956c541
commit c1b84ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/icij-worker/icij_worker/task_client.py b/icij-worker/icij_worker/task_client.py
@@ -16,14 +16,15 @@ async def create_task(
         *,
         id_: Optional[str] = None,
         group: Optional[str] = None,
-    ):
+    ) -> Task:
         if id_ is None:
             id_ = _generate_task_id(name)
         task = Task.create(task_id=id_, task_name=name, args=args)
         url = f"/api/task/{id_}"
         data = {"task": task, "group": group}
         async with self._put(url, json=data):
             pass
+        return task
 
 
 def _generate_task_id(task_name: str) -> str: