Merge pull request #278 from hypha-space/leo/fix_batch_scheduling_w_statistics

Leo/fix batch scheduling w statistics

Author: l45k

crates/messages/src/lib.rs

@@ -98,6 +98,7 @@ pub mod action {
         Idle,
         BatchCompleted {
             batch_size: u32,
+            batches: u32,
         },
         SentUpdate {
             round: u32,

crates/scheduler/src/pool.rs

@@ -913,13 +913,13 @@ mod tests {
 
         pool_with_stats.update_statistics(&peer_id, |stats, last_updated| {
             if *last_updated > 0 {
-                stats.update(10u64.saturating_sub(*last_updated));
+                stats.update(10u64.saturating_sub(*last_updated), 1);
             }
             *last_updated = 10;
         });
         pool_with_stats.update_statistics(&peer_id, |stats, last_updated| {
             if *last_updated > 0 {
-                stats.update(25u64.saturating_sub(*last_updated));
+                stats.update(25u64.saturating_sub(*last_updated), 1);
             }
             *last_updated = 25;
         });