Remove needless unwrap

huggingface · Dec 28, 2024 · 4ab3f84 · 4ab3f84
1 parent b5cbaba
commit 4ab3f84
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tokenizers/src/models/bpe/trainer.rs b/tokenizers/src/models/bpe/trainer.rs
@@ -459,11 +459,10 @@ impl BpeTrainer {
                 break;
             }
 
-            if queue.is_empty() {
+            let Some(mut top) = queue.pop() else {
                 break;
-            }
+            };
 
-            let mut top = queue.pop().unwrap();
             if top.count != pair_counts[&top.pair] as u64 {
                 top.count = pair_counts[&top.pair] as u64;
                 queue.push(top);