From bfd91c5dafe7be958f8dcaca4ac5793bf9b5c14d Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 10 Oct 2024 21:12:39 -0700 Subject: [PATCH] fix test case --- tests/unit/test_vac.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/unit/test_vac.py b/tests/unit/test_vac.py index c0d33219..4f8ead55 100644 --- a/tests/unit/test_vac.py +++ b/tests/unit/test_vac.py @@ -71,9 +71,7 @@ def check_helmets(image_path): # The function can be called with the image path result = check_helmets("/home/user/edQPXGK_workers.png")""" - expected_code = """ -Edit -import os + expected_code = """import os import numpy as np from vision_agent.tools import * from typing import *