Drain instead of clearing vec

Ivordir · Mar 21, 2024 · 3cf7968 · 3cf7968
1 parent a533a8e
commit 3cf7968
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/kmeans.rs b/src/kmeans.rs
@@ -275,11 +275,9 @@ where
         for _ in 0..(samples / BATCH) {
             batch.extend((0..BATCH).map(|_| colors[distribution.sample(rng)]));
 
-            for &color in &batch {
+            for color in batch.drain(..) {
                 self.add_sample(color);
             }
-
-            batch.clear();
         }
 
         batch.extend((0..(samples % BATCH)).map(|_| colors[distribution.sample(rng)]));