Fix lint and test.

matter-labs · Oct 16, 2024 · 0422797 · 0422797
1 parent 768e257
commit 0422797
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/contracts b/contracts
diff --git a/core/lib/protobuf_config/src/prover_autoscaler.rs b/core/lib/protobuf_config/src/prover_autoscaler.rs
@@ -98,7 +98,7 @@ impl ProtoRepr for proto::ProverAutoscalerScalerConfig {
                 let (cluster_and_gpu, max) = e.read().expect("max_provers");
                 if let Some((cluster, gpu)) = cluster_and_gpu.split_once('/') {
                     acc.entry(cluster.to_string())
-                        .or_insert_with(HashMap::new)
+                        .or_default()
                         .insert(gpu.parse().expect("max_provers/gpu"), max);
                 }
                 acc

diff --git a/prover/crates/bin/prover_autoscaler/src/global/scaler.rs b/prover/crates/bin/prover_autoscaler/src/global/scaler.rs
@@ -345,7 +345,14 @@ mod tests {
         let queuer = queuer::Queuer {
             prover_job_monitor_url: "".to_string(),
         };
-        let scaler = Scaler::new(watcher, queuer, ProverAutoscalerScalerConfig::default());
+        let scaler = Scaler::new(
+            watcher,
+            queuer,
+            ProverAutoscalerScalerConfig {
+                max_provers: HashMap::from([("foo".to_string(), HashMap::from([(Gpu::L4, 100)]))]),
+                ..Default::default()
+            },
+        );
         let got = scaler.run(
             &"prover".to_string(),
             1499,
@@ -383,6 +390,6 @@ mod tests {
             },
             3,
         )]);
-        assert!(got == want);
+        assert_eq!(got, want);
     }
 }