Skip to content

Commit f952aaa

Browse files
fix(optimization): handle None metric scores in LocalEvalSampler
1 parent 2d61cb6 commit f952aaa

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

src/google/adk/optimization/local_eval_sampler.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,11 @@ def _extract_eval_data(
289289
for eval_metric_result in per_invocation_result.eval_metric_results:
290290
eval_metric_results.append({
291291
"metric_name": eval_metric_result.metric_name,
292-
"score": round(eval_metric_result.score, 2), # accurate enough
292+
"score": (
293+
round(eval_metric_result.score, 2)
294+
if eval_metric_result.score is not None
295+
else None
296+
), # accurate enough
293297
"eval_status": eval_metric_result.eval_status.name,
294298
})
295299
per_invocation_result_dict = {

0 commit comments

Comments
 (0)