From b9faf92148c7e6432ff0411783f444b7779fa36a Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Mon, 6 Apr 2026 22:59:50 +0200 Subject: Handle environment initialization on package level --- gpt2/model_test.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'gpt2/model_test.go') diff --git a/gpt2/model_test.go b/gpt2/model_test.go index d197ccc..0072254 100644 --- a/gpt2/model_test.go +++ b/gpt2/model_test.go @@ -9,6 +9,20 @@ import ( "testing" ) +func TestMain(m *testing.M) { + if err := InitializeEnvironment(); err != nil { + log.Fatal(err) + } + + exit := m.Run() + + if err := DestroyEnvironment(); err != nil { + log.Fatal(err) + } + + os.Exit(exit) +} + func fromModel() []float32 { prompt := []int64{464, 2068, 7586, 21831, 18045, 625, 262, 16931, 3290} -- cgit v1.3.1