From 54055a6dae0588d178eb355a8c36051d1b6e98a5 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Fri, 25 Apr 2025 16:15:08 -0700 Subject: [PATCH] fix test --- sample/samplers_test.go | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sample/samplers_test.go b/sample/samplers_test.go index cdcd55d43..b720f027c 100644 --- a/sample/samplers_test.go +++ b/sample/samplers_test.go @@ -74,7 +74,6 @@ func modelHelper(t testing.TB) model.BytePairEncoding { t.Fatal(err) } - types := make([]uint32, len(vocab)) tokens := make([]string, len(vocab)) for token, id := range vocab { tokens[id] = token @@ -86,7 +85,7 @@ func modelHelper(t testing.TB) model.BytePairEncoding { ``, &model.Vocabulary{ Values: tokens, - Types: types, + Types: make([]int32, len(vocab)), Merges: merges, }, )