diff --git a/process_chunks.py b/process_chunks.py index fb91aec..a6f9e4d 100755 --- a/process_chunks.py +++ b/process_chunks.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3 import os +import concurrent.futures import datetime import numpy as np import matplotlib.pyplot as plt @@ -91,9 +92,9 @@ def process_chunk(filename): def main(): - for filename in os.listdir(INPUT_DIR): - if filename.endswith(".flac"): - process_chunk(filename) + with concurrent.futures.ProcessPoolExecutor() as executor: + files = [f for f in os.listdir(INPUT_DIR) if f.endswith(".flac")] + executor.map(process_chunk, files) if __name__ == "__main__": main() \ No newline at end of file