From b09dd9b302a87fdb0766db7983fbbfccc39a15be Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 10 Oct 2024 21:36:33 -0700 Subject: [PATCH] fix format issue --- tests/unit/test_va.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/unit/test_va.py b/tests/unit/test_va.py index 8f5e234d..3fe619e8 100644 --- a/tests/unit/test_va.py +++ b/tests/unit/test_va.py @@ -30,9 +30,7 @@ def test_parse_execution_no_test_multi_plan_edit(): def test_parse_execution_custom_tool_names_generate(): code = "generate_vision_code(artifacts, 'code.py', 'Generate code', ['image.png'])" assert ( - parse_execution( - code, test_multi_plan=False, custom_tool_names=["owl_v2_image"] - ) + parse_execution(code, test_multi_plan=False, custom_tool_names=["owl_v2_image"]) == "generate_vision_code(artifacts, 'code.py', 'Generate code', ['image.png'], test_multi_plan=False, custom_tool_names=['owl_v2_image'])" ) @@ -40,9 +38,7 @@ def test_parse_execution_custom_tool_names_generate(): def test_parse_execution_custom_tool_names_edit(): code = "edit_vision_code(artifacts, 'code.py', ['Generate code'], ['image.png'])" assert ( - parse_execution( - code, test_multi_plan=False, custom_tool_names=["owl_v2_image"] - ) + parse_execution(code, test_multi_plan=False, custom_tool_names=["owl_v2_image"]) == "edit_vision_code(artifacts, 'code.py', ['Generate code'], ['image.png'], test_multi_plan=False, custom_tool_names=['owl_v2_image'])" )