Skip to content

Commit faecc3b

Browse files
committed
cargo fmt
1 parent 5322c28 commit faecc3b

File tree

1 file changed

+19
-7
lines changed

1 file changed

+19
-7
lines changed

src/bpe/mod.rs

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
//! b-p-e for Byte Pair Encoding
1+
//! b-p-e for Byte Pair Encoding
22
33
mod algorithm;
44

@@ -283,8 +283,8 @@ mod bpe_tests {
283283

284284
let encoded: Vec<_> = bpe.encode("abc").into_iter().collect();
285285
assert_eq!(encoded.len(), 2); // Should merge "ab" and leave "c"
286-
assert_eq!(encoded[0], 3); // Assuming "ab" is assigned token ID 3
287-
assert_eq!(encoded[1], 2); // Assuming "c" is assigned token ID 2
286+
assert_eq!(encoded[0], 3); // Assuming "ab" is assigned token ID 3
287+
assert_eq!(encoded[1], 2); // Assuming "c" is assigned token ID 2
288288
}
289289

290290
#[test]
@@ -307,7 +307,10 @@ mod bpe_tests {
307307

308308
let text = "abcbc";
309309
let encoded: Vec<_> = bpe.encode(text).into_iter().collect();
310-
let decoded: Vec<u8> = encoded.iter().flat_map(|&t| bpe.decode(t).iter().copied()).collect();
310+
let decoded: Vec<u8> = encoded
311+
.iter()
312+
.flat_map(|&t| bpe.decode(t).iter().copied())
313+
.collect();
311314
assert_eq!(String::from_utf8(decoded).unwrap(), text);
312315
}
313316

@@ -333,13 +336,22 @@ mod bpe_tests {
333336
println!("Inaccessible tokens: {:?}", inaccessible);
334337

335338
// 'd' is a single character, so it should be accessible
336-
assert!(!inaccessible.contains_key("d"), "Token 'd' should be accessible");
339+
assert!(
340+
!inaccessible.contains_key("d"),
341+
"Token 'd' should be accessible"
342+
);
337343

338344
// 'bcd' cannot be formed by merging other tokens, so it should be inaccessible
339-
assert!(inaccessible.contains_key("bcd"), "Token 'bcd' should be inaccessible");
345+
assert!(
346+
inaccessible.contains_key("bcd"),
347+
"Token 'bcd' should be inaccessible"
348+
);
340349

341350
// 'ab' can be formed by merging 'a' and 'b', so it should be accessible
342-
assert!(!inaccessible.contains_key("ab"), "Token 'ab' should be accessible");
351+
assert!(
352+
!inaccessible.contains_key("ab"),
353+
"Token 'ab' should be accessible"
354+
);
343355
}
344356

345357
#[test]

0 commit comments

Comments
 (0)