Fix compat

Author: mfranciszkiewicz

examples/blender/blender.py

@@ -34,7 +34,7 @@ async def worker(ctx: WorkContext, tasks):
             )
             ctx.run("/golem/entrypoints/run-blender.sh")
             ctx.download_file(f"/golem/output/out{frame:04d}.png", f"output_{frame}.png")
-            yield task, ctx.commit()
+            yield ctx.commit(task)
             # TODO: Check if job results are valid
             # and reject by: task.reject_task(msg = 'invalid file')
             task.accept_task()

yapapi/runner/__init__.py

@@ -157,7 +157,7 @@ def __init__(
         self._budget_allocation: Optional[rest.payment.Allocation] = None
 
     async def map(
-        self, worker: Callable[[WorkContext, AsyncIterator["Task"]], AsyncIterator[Tuple["Task", Work]]], data
+        self, worker: Callable[[WorkContext, AsyncIterator["Task"]], AsyncIterator[Work]], data
     ):
         import asyncio
         import contextlib

yapapi/runner/ctx.py

@@ -173,7 +173,7 @@ def download_file(self, src_path: str, dst_path: str):
     def log(self, *args):
         print(f"W{self._id}: ", *args)
 
-    def commit(self) -> Work:
+    def commit(self, task: "Task") -> Tuple["Task", Work]:
         steps = self._pending_steps
         self._pending_steps = []
-        return _Steps(*steps)
+        return task, _Steps(*steps)