Remove unnecessary Finalize function

nlpodyssey · Oct 30, 2023 · 098ac25 · 098ac25
1 parent 3c0cb00
commit 098ac25
Show file tree

Hide file tree

Showing 11 changed files with 0 additions and 33 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -72,7 +72,6 @@ func run() error {
 	if err != nil {
 		return err
 	}
-	defer tasks.Finalize(m)
 
 	logMetrics()
 

diff --git a/examples/abstractivequestionasnwering/main.go b/examples/abstractivequestionasnwering/main.go
@@ -42,7 +42,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	opts := textgeneration.DefaultOptions()
 

diff --git a/examples/languagemodeling/main.go b/examples/languagemodeling/main.go
@@ -28,7 +28,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	fn := func(text string) error {
 		result, err := m.Predict(context.Background(), text, languagemodeling.Parameters{K: 10})

diff --git a/examples/questionanswering/main.go b/examples/questionanswering/main.go
@@ -32,7 +32,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	opts := &questionanswering.Options{}
 

diff --git a/examples/relationextraction/main.go b/examples/relationextraction/main.go
@@ -36,7 +36,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	opts := textgeneration.DefaultOptions()
 

diff --git a/examples/textclassification/main.go b/examples/textclassification/main.go
@@ -31,7 +31,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	fn := func(text string) error {
 		start := time.Now()

diff --git a/examples/textencoding/main.go b/examples/textencoding/main.go
@@ -31,7 +31,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	fn := func(text string) error {
 		result, err := m.Encode(context.Background(), text, int(bert.MeanPooling))

diff --git a/examples/textgeneration/main.go b/examples/textgeneration/main.go
@@ -34,7 +34,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	log.Debug().Msgf("Loaded model %q in %v", modelName, time.Since(start))
 

diff --git a/examples/tokenclassification/main.go b/examples/tokenclassification/main.go
@@ -29,7 +29,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	params := tokenclassification.Parameters{
 		AggregationStrategy: tokenclassification.AggregationStrategySimple,

diff --git a/examples/zeroshotclassification/main.go b/examples/zeroshotclassification/main.go
@@ -35,7 +35,6 @@ func main() {
 	if err != nil {
 		log.Fatal().Err(err).Send()
 	}
-	defer tasks.Finalize(m)
 
 	params := zeroshotclassifier.Parameters{
 		CandidateLabels:    strings.Split(possibleClasses, ","),

diff --git a/pkg/tasks/finalizer.go b/pkg/tasks/finalizer.go