fix cancel

edenartlab · Jan 14, 2025 · e65f9df · e65f9df
1 parent e86ba52
commit e65f9df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/eve/postprocessing.py b/eve/postprocessing.py
@@ -19,7 +19,7 @@
 from eve.models import Model
 
 
-def cancel_stuck_tasks():
+async def cancel_stuck_tasks():
     tasks = get_collection(Task.collection_name)
 
     expired_tasks = tasks.find({
@@ -43,7 +43,7 @@ def cancel_stuck_tasks():
 
         try:    
             tool = Tool.load(key=task.tool)
-            tool.cancel(task, force=True)
+            await tool.async_cancel(task, force=True)
 
         except Exception as e:
             print(f"Error canceling task {str(task.id)} {task.tool}", e)

diff --git a/remote_replicate_tool.py b/remote_replicate_tool.py
@@ -24,6 +24,7 @@
     modal.Image.debian_slim(python_version="3.11")
     .apt_install("libmagic1", "ffmpeg", "wget")
     .pip_install_from_pyproject(str(root_dir / "pyproject.toml"))
+    .run_commands(["playwright install"])
     .env({"DB": db})
 )