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