8000 fix eval unittests failure · calvingiles/adk-python@5462862 · GitHub
[go: up one dir, main page]

Skip to content

Commit 5462862

Browse files
seanzhougooglecopybara-github
authored andcommitted
fix eval unittests failure
PiperOrigin-RevId: 757872670
1 parent c39f24f commit 5462862

File tree

2 files changed

+263
-213
lines changed

2 files changed

+263
-213
lines changed

src/google/adk/cli/cli_tools_click.py

Lines changed: 12 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -269,25 +269,20 @@ def cli_eval(
269269

270270
eval_set_to_evals = parse_and_get_evals_to_run(eval_set_file_path)
271271

272-
async def _collect_async_gen(
273-
async_gen_coroutine: Coroutine[
274-
AsyncGenerator[EvalResult, None], None, None
275-
],
276-
) -> list[EvalResult]:
277-
return [result async for result in async_gen_coroutine]
272+
async def _collect_eval_results() -> list[EvalResult]:
273+
return [
274+
result
275+
async for result in run_evals(
276+
eval_set_to_evals,
277+
root_agent,
278+
reset_func,
279+
eval_metrics,
280+
print_detailed_results=print_detailed_results,
281+
)
282+
]
278283

279284
try:
280-
eval_results = asyncio.run(
281-
_collect_async_gen(
282-
run_evals(
283-
eval_set_to_evals,
284-
root_agent,
285-
reset_func,
286-
eval_metrics,
287-
print_detailed_results=print_detailed_results,
288-
)
289-
)
290-
)
285+
eval_results = asyncio.run(_collect_eval_results())
291286
except ModuleNotFoundError:
292287
raise click.ClickException(MISSING_EVAL_DEPENDENCIES_MESSAGE)
293288

0 commit comments

Comments
 (0)
0