set a 15 sec timeout for getting initial download progress

6855a7727d08440ac5922099df424642d2e897dc · 2025-08-03 20:37:20 +0800 · Alex Cheema

Files touched

M worker/worker.py

Diff

commit 6855a7727d08440ac5922099df424642d2e897dc
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sun Aug 3 20:37:20 2025 +0800

    set a 15 sec timeout for getting initial download progress
---
 worker/worker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/worker/worker.py b/worker/worker.py
index 8e388d93..6ac3f47c 100644
--- a/worker/worker.py
+++ b/worker/worker.py
@@ -171,7 +171,7 @@ class Worker:
         This op assigns the runner, and moves from Downloading -> Inactive (ready to spin) state.
         """
         assigned_runner = self._create_assigned_runner(op)
-        initial_progress = await self.shard_downloader.get_shard_download_status_for_shard(op.shard_metadata)
+        initial_progress = await asyncio.wait_for(self.shard_downloader.get_shard_download_status_for_shard(op.shard_metadata), timeout=15)
 
         if initial_progress.status == "complete":
             async for event in self._handle_already_downloaded_shard(assigned_runner):

← 1fe4ed34 Worker Exception & Timeout Refactor · back to Exo · run configure_mlx.sh in run.sh f352ddfc →