fix

knights-analytics · Mar 21, 2024 · 74f8929 · 74f8929
1 parent bb77090
commit 74f8929
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cmd/main_test.go b/cmd/main_test.go
@@ -19,14 +19,13 @@ import (
 var textClassificationData []byte
 
 //go:embed testData/tokenClassification.jsonl
-// var tokenClassificationData []byte
+var tokenClassificationData []byte
 
 func TestMain(m *testing.M) {
 	// model setup
 	if ok, err := util.FileSystem.Exists(context.Background(), "/build/models"); err == nil {
-		fmt.Println("HERE NO ERROR")
+		fmt.Println("HERE")
 		if !ok {
-			fmt.Println("HERE MODEL FOLDER NOT THERE")
 			session, err := hugot.NewSession()
 			if err != nil {
 				panic(err)