Merge pull request #277 from hypha-space/leo/fix_push_streaming_retry

Fix push streaming retry

Author: l45k

crates/worker/src/connector/mod.rs

@@ -353,7 +353,7 @@ where
                                 let network = network.clone();
                                 async move {
                                     let retry_strategy =
-                                        FixedInterval::from_millis(200).map(jitter).take(6);
+                                        FixedInterval::from_millis(200).map(jitter).take(15);
 
                                     async fn attempt_push<T>(
                                         network: T,

crates/worker/src/executor/parameter_server.rs

@@ -340,18 +340,12 @@ impl JobExecutor for ParameterServerExecutor {
                                     }
                                 };
 
-                                match Retry::spawn(retry_strategy.clone(), || {
-                                    let connector = connector.clone();
-                                    let send = send.clone();
-                                    let cancel = cancel.clone();
-                                    async move { broadcast_update(
+                                match broadcast_update(
                                     connector.clone(),
                                     send,
                                     gradient_file,
-                                    cancel.clone(),
+                                    cancel.clone()
                                 ).await
-                                    }
-                                }).await
                                 {
                                     Ok(()) => {
                                         pending_update = None;