From c30621652058dae1b986fa7ad225ab634cbacf0a Mon Sep 17 00:00:00 2001 From: Minh Vu Date: Sun, 24 May 2026 13:15:42 +0200 Subject: [PATCH] Remove unused generic from CoreBPE::new --- src/lib.rs | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/src/lib.rs b/src/lib.rs index ea54eac8..e7c6ede7 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -598,7 +598,7 @@ impl CoreBPE { (tokens, completions) } - pub fn new( + pub fn new( encoder: E, special_tokens_encoder: SE, pattern: &str, @@ -606,7 +606,6 @@ impl CoreBPE { where E: IntoIterator, Rank)>, SE: IntoIterator, - NSE: IntoIterator, { Self::new_internal( HashMap::from_iter(encoder), @@ -677,10 +676,9 @@ impl CoreBPE { #[cfg(test)] mod tests { - use fancy_regex::Regex; use rustc_hash::FxHashMap as HashMap; - use crate::{Rank, byte_pair_split}; + use crate::{CoreBPE, Rank, byte_pair_split}; fn setup_ranks() -> HashMap, Rank> { HashMap::from_iter([(b"ab".to_vec(), 0), (b"cd".to_vec(), 1)]) @@ -699,4 +697,16 @@ mod tests { let res = byte_pair_split(b"abab", &ranks); assert_eq!(res, vec![b"ab", b"ab"]); } + + #[test] + fn test_public_constructor_is_callable() { + let bpe = CoreBPE::new( + vec![(b"ab".to_vec(), 0)], + Vec::<(String, Rank)>::new(), + r".+", + ) + .unwrap(); + + assert_eq!(bpe.encode_ordinary("ab"), vec![0]); + } }