Commit 40fb07c6 authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

[fix] properly update metadata list

parent d022397e

ivas_processing_scripts/processing/processing.py

+5 −8
Original line number Diff line number Diff line
@@ -52,7 +52,6 @@ from ivas_processing_scripts.constants import LOGGER_DATEFMT, LOGGER_FORMAT 
from ivas_processing_scripts.processing.config import TestConfig

from ivas_processing_scripts.processing.tx import get_timescaled_splits

from ivas_processing_scripts.utils import (

    ALLOWED_INPUT_EXT_MD,

    apply_func_parallel,

    list_audio_or_md,

    pairwise,
@@ -415,13 +414,11 @@ def preprocess_2(cfg, logger): 
        cfg.out_dirs[0], select_list=getattr(cfg, "input_select", None)

    )



    cfg.metadata_path = [

        list_audio_or_md(

            cfg.out_dirs[0],

            select_list=getattr(cfg, "input_select", None),

            allowed_ext=ALLOWED_INPUT_EXT_MD,

        )

    ]

    # update the metadata list to use preprocessing 2 outputs as well

    for md_files in cfg.metadata_path:

        for i, md in enumerate(md_files):

            if md:

                md_files[i] = cfg.out_dirs[0] / md.stem



    # remove already applied processing stage

    cfg.proc_chains = cfg.proc_chains[1:]