From 1038aa7d8167d8089525440447d806318d21ba4f Mon Sep 17 00:00:00 2001 From: danellecline Date: Wed, 23 Oct 2024 17:19:19 -0700 Subject: [PATCH] fix(bio): removed unused args --- aipipeline/projects/bio/run_strided_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aipipeline/projects/bio/run_strided_inference.py b/aipipeline/projects/bio/run_strided_inference.py index 9888afe..25224ae 100644 --- a/aipipeline/projects/bio/run_strided_inference.py +++ b/aipipeline/projects/bio/run_strided_inference.py @@ -387,13 +387,13 @@ def seconds_to_timestamp(seconds): jpg_file.unlink() -def process_videos(video_files, stride, endpoint_url, class_name, version_id, min_confidence, +def process_videos(video_files, stride, endpoint_url, version_id, min_confidence, allowed_classes, class_remap, remove_vignette=False, skip_vss=False): num_cpus = multiprocessing.cpu_count() pool = multiprocessing.Pool(processes=num_cpus) pool.starmap( run_inference, - [(v, stride, endpoint_url, class_name, version_id, min_confidence, allowed_classes, class_remap, remove_vignette, skip_vss) for v in + [(v, stride, endpoint_url, version_id, min_confidence, allowed_classes, class_remap, remove_vignette, skip_vss) for v in video_files], ) pool.close()