Merge branch 'feature-hw' of https://github.com/killerdbob/skyplane i…

…nto feature-hw
skyplane-project · Aug 14, 2023 · 9a3f2f7 · 9a3f2f7
2 parents c3b6f91 + f78bd7c
commit 9a3f2f7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/skyplane/gateway/operators/gateway_operator.py b/skyplane/gateway/operators/gateway_operator.py
@@ -90,7 +90,14 @@ def worker_loop(self, worker_id: int, *args):
                 # TODO: status logging
                 self.chunk_store.log_chunk_state(chunk_req, ChunkState.in_progress, operator_handle=self.handle, worker_id=worker_id)
                 # process chunk
-                succ = self.process(chunk_req, *args)
+                succ = retry_backoff(
+                    partial(
+                        self.process,
+                        chunk_req, 
+                        *args
+                    ),
+                    max_retries=1,
+                )
 
                 # place in output queue
                 if succ: