diff --git a/src/google/adk/evaluation/local_eval_service.py b/src/google/adk/evaluation/local_eval_service.py index ab285e1836..09a08d37bb 100644 --- a/src/google/adk/evaluation/local_eval_service.py +++ b/src/google/adk/evaluation/local_eval_service.py @@ -130,7 +130,7 @@ async def perform_inference( async def run_inference(eval_case): async with semaphore: - return await self._perform_inference_sigle_eval_item( + return await self._perform_inference_single_eval_item( app_name=inference_request.app_name, eval_set_id=inference_request.eval_set_id, eval_case=eval_case, @@ -383,7 +383,7 @@ def _generate_final_eval_status( return final_eval_status - async def _perform_inference_sigle_eval_item( + async def _perform_inference_single_eval_item( self, app_name: str, eval_set_id: str, diff --git a/tests/unittests/evaluation/test_local_eval_service.py b/tests/unittests/evaluation/test_local_eval_service.py index 8d973c92da..df90bf39dc 100644 --- a/tests/unittests/evaluation/test_local_eval_service.py +++ b/tests/unittests/evaluation/test_local_eval_service.py @@ -174,7 +174,7 @@ async def test_perform_inference_success( mock_eval_sets_manager.get_eval_set.return_value = eval_set mock_inference_result = mocker.MagicMock() - eval_service._perform_inference_sigle_eval_item = mocker.AsyncMock( + eval_service._perform_inference_single_eval_item = mocker.AsyncMock( return_value=mock_inference_result ) @@ -194,7 +194,7 @@ async def test_perform_inference_success( mock_eval_sets_manager.get_eval_set.assert_called_once_with( app_name="test_app", eval_set_id="test_eval_set" ) - assert eval_service._perform_inference_sigle_eval_item.call_count == 2 + assert eval_service._perform_inference_single_eval_item.call_count == 2 @pytest.mark.asyncio @@ -215,7 +215,7 @@ async def test_perform_inference_with_case_ids( mock_eval_sets_manager.get_eval_set.return_value = eval_set mock_inference_result = mocker.MagicMock() - eval_service._perform_inference_sigle_eval_item = mocker.AsyncMock( + eval_service._perform_inference_single_eval_item = mocker.AsyncMock( return_value=mock_inference_result ) @@ -231,13 +231,13 @@ async def test_perform_inference_with_case_ids( results.append(result) assert len(results) == 2 - eval_service._perform_inference_sigle_eval_item.assert_any_call( + eval_service._perform_inference_single_eval_item.assert_any_call( app_name="test_app", eval_set_id="test_eval_set", eval_case=eval_set.eval_cases[0], root_agent=dummy_agent, ) - eval_service._perform_inference_sigle_eval_item.assert_any_call( + eval_service._perform_inference_single_eval_item.assert_any_call( app_name="test_app", eval_set_id="test_eval_set", eval_case=eval_set.eval_cases[2],