diff --git a/tests/unittests/evaluation/test_local_eval_sets_manager.py b/tests/unittests/evaluation/test_local_eval_sets_manager.py index 3450fb9338..fd05a6c032 100644 --- a/tests/unittests/evaluation/test_local_eval_sets_manager.py +++ b/tests/unittests/evaluation/test_local_eval_sets_manager.py @@ -624,7 +624,7 @@ def test_local_eval_sets_manager_update_eval_case_eval_set_not_found( updated_eval_case = EvalCase(eval_id=eval_case_id, conversation=[]) mocker.patch( - "google.adk.evaluation.local_eval_sets_manager.LocalEvalSetsManager.get_eval_case", + "google.adk.evaluation.local_eval_sets_manager.LocalEvalSetsManager.get_eval_set", return_value=None, ) @@ -708,7 +708,7 @@ def test_local_eval_sets_manager_delete_eval_case_eval_set_not_found( eval_case_id = "test_eval_case" mocker.patch( - "google.adk.evaluation.local_eval_sets_manager.LocalEvalSetsManager.get_eval_case", + "google.adk.evaluation.local_eval_sets_manager.LocalEvalSetsManager.get_eval_set", return_value=None, ) mock_write_eval_set_to_path = mocker.patch(