Skip to content

Commit

Permalink
fix(bio): removed unused args
Browse files Browse the repository at this point in the history
  • Loading branch information
danellecline committed Oct 24, 2024
1 parent 68707a4 commit 1038aa7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aipipeline/projects/bio/run_strided_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit 1038aa7

Please sign in to comment.