Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
riccardopinosio committed Mar 21, 2024
1 parent 1181ba6 commit 078b0bd
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions cmd/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,15 +22,6 @@ var textClassificationData []byte
var tokenClassificationData []byte

func TestMain(m *testing.M) {
entries, err := os.ReadDir("../models")
if err != nil {
log.Fatal(err)
}

for _, e := range entries {
fmt.Println(e.Name())
}

// model setup
if ok, err := util.FileSystem.Exists(context.Background(), "../models"); err == nil {
fmt.Println("HERE")
Expand Down Expand Up @@ -76,6 +67,17 @@ func TestTextClassificationCli(t *testing.T) {

testModel := path.Join("../models", "KnightsAnalytics_distilbert-base-uncased-finetuned-sst-2-english")

fmt.Println("TESTMODEL", testModel)

entries, err := os.ReadDir("../models")
if err != nil {
log.Fatal(err)
}

for _, e := range entries {
fmt.Println(e.Name())
}

// write the test data and test recursive reads and processing from folder
testDataDir := path.Join(os.TempDir(), "hugoTestData")
recurseDir := path.Join(testDataDir, "cliRecurseTest")
Expand Down

0 comments on commit 078b0bd

Please sign in to comment.