fix format issue

landing-ai · Oct 11, 2024 · b09dd9b · b09dd9b
1 parent 7f1aba9
commit b09dd9b
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/unit/test_va.py b/tests/unit/test_va.py
@@ -30,19 +30,15 @@ def test_parse_execution_no_test_multi_plan_edit():
 def test_parse_execution_custom_tool_names_generate():
  code = "<execute_python>generate_vision_code(artifacts, 'code.py', 'Generate code', ['image.png'])</execute_python>"
  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'])"
  )
 
 
 def test_parse_execution_custom_tool_names_edit():
  code = "<execute_python>edit_vision_code(artifacts, 'code.py', ['Generate code'], ['image.png'])</execute_python>"
  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'])"
  )