From 74a331f1fb90e85c830223b05673f59de1873307 Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Sun, 25 May 2025 20:32:49 -0700 Subject: [PATCH] chore: reformat the code with autoformat.sh PiperOrigin-RevId: 763256626 --- .../tools/openapi_tool/openapi_spec_parser/rest_api_tool.py | 2 +- tests/unittests/cli/utils/test_cli_tools_click.py | 3 +-- tests/unittests/sessions/test_vertex_ai_session_service.py | 4 ++-- 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/src/google/adk/tools/openapi_tool/openapi_spec_parser/rest_api_tool.py b/src/google/adk/tools/openapi_tool/openapi_spec_parser/rest_api_tool.py index 7396255..0491805 100644 --- a/src/google/adk/tools/openapi_tool/openapi_spec_parser/rest_api_tool.py +++ b/src/google/adk/tools/openapi_tool/openapi_spec_parser/rest_api_tool.py @@ -48,7 +48,7 @@ _OPENAPI_SCHEMA_IGNORE_FIELDS = ( "format", "additional_properties", "ref", - "def" + "def", ) diff --git a/tests/unittests/cli/utils/test_cli_tools_click.py b/tests/unittests/cli/utils/test_cli_tools_click.py index cdd19a9..b70e168 100644 --- a/tests/unittests/cli/utils/test_cli_tools_click.py +++ b/tests/unittests/cli/utils/test_cli_tools_click.py @@ -302,8 +302,7 @@ def test_cli_eval_success_path( stub.try_get_reset_func = lambda _p: None stub.parse_and_get_evals_to_run = lambda _paths: {"set1.json": ["e1", "e2"]} eval_sets_manager_stub.load_eval_set_from_file = lambda x, y: EvalSet( - "test_eval_set_id", - [EvalCase("e1"), EvalCase("e2")] + "test_eval_set_id", [EvalCase("e1"), EvalCase("e2")] ) # Create an async generator function for run_evals diff --git a/tests/unittests/sessions/test_vertex_ai_session_service.py b/tests/unittests/sessions/test_vertex_ai_session_service.py index 83351bc..8144cfe 100644 --- a/tests/unittests/sessions/test_vertex_ai_session_service.py +++ b/tests/unittests/sessions/test_vertex_ai_session_service.py @@ -214,9 +214,9 @@ def mock_get_api_client(): '1': MOCK_EVENT_JSON, } with mock.patch( - "google.adk.sessions.vertex_ai_session_service._get_api_client", + 'google.adk.sessions.vertex_ai_session_service._get_api_client', return_value=api_client, - ): + ): yield