Commit 21c6f5c5 authored by Anika Treffehn's avatar Anika Treffehn
Browse files

formatting

parent 76f97e33

ivas_processing_scripts/processing/chains.py

+13 −5
Original line number Diff line number Diff line
@@ -30,8 +30,8 @@ 
#  the United Nations Convention on Contracts on the International Sales of Goods.

#



from typing import Optional

from pathlib import Path

from typing import Optional



from ivas_processing_scripts.processing.config import TestConfig

from ivas_processing_scripts.processing.evs import EVS
@@ -137,7 +137,9 @@ def get_preprocessing_2(cfg: TestConfig) -> dict: 


        background = {

            "snr": background_cfg.get("snr", None),

            "background_noise_path": get_abs_path(background_cfg.get("background_noise_path", None)),

            "background_noise_path": get_abs_path(

                background_cfg.get("background_noise_path", None)

            ),

            "seed_delay": background_cfg.get("seed_delay", 0),

            "master_seed": cfg.master_seed,

            "output_fmt": cfg.postprocessing["fmt"],
@@ -252,7 +254,9 @@ def get_processing_chain( 
            if tx_cfg_tmp.get("type", None) == "FER":

                tx_cfg = {

                    "type": tx_cfg_tmp.get("type", None),

                    "error_pattern": get_abs_path(tx_cfg_tmp.get("error_pattern", None)),

                    "error_pattern": get_abs_path(

                        tx_cfg_tmp.get("error_pattern", None)

                    ),

                    "error_rate": tx_cfg_tmp.get("error_rate", None),

                    "master_seed": cfg.master_seed,

                    "prerun_seed": tx_cfg_tmp.get("prerun_seed", 0),
@@ -260,7 +264,9 @@ def get_processing_chain( 
            elif tx_cfg_tmp.get("type", None) == "JBM":

                tx_cfg = {

                    "type": tx_cfg_tmp.get("type", None),

                    "error_pattern": get_abs_path(tx_cfg_tmp.get("error_pattern", None)),

                    "error_pattern": get_abs_path(

                        tx_cfg_tmp.get("error_pattern", None)

                    ),

                    "error_profile": tx_cfg_tmp.get("error_profile", None),

                    "n_frames_per_packet": tx_cfg_tmp.get("n_frames_per_packet", None),

                }
@@ -311,7 +317,9 @@ def get_processing_chain( 
            if tx_cfg_tmp.get("type", None) == "FER":

                tx_cfg = {

                    "type": tx_cfg_tmp.get("type", None),

                    "error_pattern": get_abs_path(tx_cfg_tmp.get("error_pattern", None)),

                    "error_pattern": get_abs_path(

                        tx_cfg_tmp.get("error_pattern", None)

                    ),

                    "error_rate": tx_cfg_tmp.get("error_rate", None),

                    "master_seed": cfg.master_seed,

                    "prerun_seed": tx_cfg_tmp.get("prerun_seed", 0),

ivas_processing_scripts/utils.py

+3 −1
Original line number Diff line number Diff line
@@ -106,7 +106,9 @@ def list_audio(path: str, select_list: list = None) -> list: 
    if path.exists():

        if path.is_dir():

            audio_list = [

                f.resolve().absolute() for f in path.iterdir() if f.suffix in ALLOWED_INPUT_EXT

                f.resolve().absolute()

                for f in path.iterdir()

                if f.suffix in ALLOWED_INPUT_EXT

            ]

        else:

            ext = path.suffix