Skip to content
Snippets Groups Projects
Commit dc92b137 authored by Grießhaber Daniel's avatar Grießhaber Daniel :squid:
Browse files

fix backend task instantiation

parent c69e1629
No related branches found
No related tags found
No related merge requests found
......@@ -10,7 +10,7 @@ from evoprompt.api import config
from evoprompt.cli import argument_parser
from evoprompt.evolution import GeneticAlgorithm
from evoprompt.models import Llama, LLMModel
from evoprompt.task import SentimentAnalysis
from evoprompt.task.sentiment_analysis import SST2
class MultiProcessOptimizer:
......@@ -34,7 +34,7 @@ class MultiProcessOptimizer:
[
"--debug" if debug else "",
"--task",
SentimentAnalysis.shorthand,
SST2.shorthand,
]
)
......@@ -55,7 +55,7 @@ class MultiProcessOptimizer:
).result()
# currently fix task
task = SentimentAnalysis(evaluation_model, self.options)
task = SST2(evaluation_model, **vars(self.options))
optimizer_class = GeneticAlgorithm
# optimizer_class = DifferentialEvolution
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment