Skip to content

Commit c30334f

Browse files
author
Ismail Mehdi
committed
fix normal flow
1 parent 06e066a commit c30334f

3 files changed

Lines changed: 9 additions & 4 deletions

File tree

evalbench/eval_service.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010
import grpc
1111
import pathlib
1212
from dataset.dataset import load_json
13-
from evaluator.orchestrator import Orchestrator
13+
from evaluator import get_orchestrator
14+
1415
import reporting.report as report
1516
from reporting import get_reporters
1617
import reporting.analyzer as analyzer
@@ -114,7 +115,10 @@ async def Eval(
114115
config, db_configs, model_config, setup_config = load_session_configs(session)
115116
dataset = await get_dataset_from_request(request_iterator)
116117

117-
evaluator = Orchestrator(config, db_configs, setup_config)
118+
evaluator = get_orchestrator(
119+
config, db_configs, setup_config, report_progress=True
120+
)
121+
118122
evaluator.evaluate(dataset)
119123

120124
job_id, run_time, results_tf, scores_tf = evaluator.process()

evalbench/generators/prompts/passthrough.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,5 @@ def setup(self):
99
pass
1010

1111
def generate(self, item):
12+
item["prompt"] = item["nl_prompt"]
1213
return item

evalbench/work/promptgenwork.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ def run(self, work_config: Any = None) -> dict:
2323
generated_prompt = None
2424
prompt_generator_error = None
2525
try:
26-
generated_prompt = self.generator.generate(self.eval_result)
26+
self.generator.generate(self.eval_result)
2727
except Exception as e:
2828
prompt_generator_error = str(e)
2929

30-
self.eval_result["generated_prompt"] = generated_prompt
30+
self.eval_result["generated_prompt"] = self.eval_result["prompt"]
3131
self.eval_result["prompt_generator_error"] = prompt_generator_error
3232
return self.eval_result

0 commit comments

Comments
 (0)