diff --git a/llama_stack/providers/tests/scoring/test_scoring.py b/llama_stack/providers/tests/scoring/test_scoring.py index 503ccedb3..2c6ff1641 100644 --- a/llama_stack/providers/tests/scoring/test_scoring.py +++ b/llama_stack/providers/tests/scoring/test_scoring.py @@ -54,3 +54,13 @@ class TestScoring: for x in scoring_functions: assert x in response.results assert len(response.results[x].score_rows) == len(rows.rows) + + # score batch + response = await scoring_impl.score_batch( + dataset_id="test_dataset", + scoring_functions=scoring_functions, + ) + assert len(response.results) == len(scoring_functions) + for x in scoring_functions: + assert x in response.results + assert len(response.results[x].score_rows) == 5