fix one test?

This commit is contained in:
Arthur Zucker
2023-09-01 18:42:30 +00:00
parent 345b4eba96
commit 6cca5716af

View File

@ -650,7 +650,7 @@ mod tests {
assert_eq!(*vocab.get_vocab_r(), HashMap::from([ assert_eq!(*vocab.get_vocab_r(), HashMap::from([
(2,AddedToken::from("added_token_1", true)), (2,AddedToken::from("added_token_1", true)),
(3,AddedToken::from("added_token_2", true)), (3,AddedToken::from("added_token_2", true)),
(4,AddedToken::from("test", true)), (0,AddedToken::from("test", true)),
])); ]));
assert!(!vocab.added_tokens_map.contains_key("test")); assert!(!vocab.added_tokens_map.contains_key("test"));