From fa74ae72142be7683ed968cc2f5a11c5935e373a Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Fri, 31 Jan 2025 14:55:09 -0800 Subject: [PATCH] use range --- model/process_text_benchmark_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/model/process_text_benchmark_test.go b/model/process_text_benchmark_test.go index aa7185094..f5cc38089 100644 --- a/model/process_text_benchmark_test.go +++ b/model/process_text_benchmark_test.go @@ -81,7 +81,7 @@ func BenchmarkBytePairEncoding(b *testing.B) { // Benchmark Encoding b.Run("Encode_"+bm.name, func(b *testing.B) { b.ReportAllocs() - for i := 0; i < b.N; i++ { + for range b.N { tokens, err := bpe.Encode(bm.input) if err != nil { b.Fatal(err) @@ -99,7 +99,7 @@ func BenchmarkBytePairEncoding(b *testing.B) { // Benchmark Decoding b.Run("Decode_"+bm.name, func(b *testing.B) { b.ReportAllocs() - for i := 0; i < b.N; i++ { + for range b.N { decoded, err := bpe.Decode(tokens) if err != nil { b.Fatal(err) @@ -148,7 +148,7 @@ func BenchmarkBytePairEncodingSplit(b *testing.B) { for _, bm := range benchmarks { b.Run("Split_"+bm.name, func(b *testing.B) { b.ReportAllocs() - for i := 0; i < b.N; i++ { + for range b.N { splits, err := bpe.split(bm.input) if err != nil { b.Fatal(err)