Skip to content

Commit

Permalink
fix(bio): correct clean-up
Browse files Browse the repository at this point in the history
  • Loading branch information
danellecline committed Oct 12, 2024
1 parent 0531405 commit 7c2684c
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion aipipeline/projects/bio/run_strided_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,9 @@ def seconds_to_timestamp(seconds):
logger.error(f"Error processing frame at {current_time_secs} seconds: {response.text}")

logger.info(f"Finished processing video {video_path}")
output_path.unlink(missing_ok=True)
# Remove the directory with the frames
for jpg_file in output_path.glob("*.jpg"):
jpg_file.unlink()


def process_videos(video_files, stride, endpoint_url, config_dict, class_name, version_id):
Expand Down

0 comments on commit 7c2684c

Please sign in to comment.