run formatter (1b63ada4) · Commits · IVAS Codec Public Collaboration / IVAS Processing Scripts

ivas_processing_scripts/processing/chains.py

+3 −3

Original line number	Diff line number	Diff line
		@@ -33,6 +33,8 @@
		from typing import Optional
		from warnings import warn

		from ivas_processing_scripts.audiotools import audio
		from ivas_processing_scripts.audiotools.audiofile import read
		from ivas_processing_scripts.processing.config import TestConfig
		from ivas_processing_scripts.processing.evs import EVS
		from ivas_processing_scripts.processing.ivas import IVAS
		@@ -43,8 +45,6 @@ from ivas_processing_scripts.processing.processing_splitting_scaling import (
		Processing_splitting_scaling,
		)
		from ivas_processing_scripts.utils import get_abs_path, list_audio
		from ivas_processing_scripts.audiotools import audio
		from ivas_processing_scripts.audiotools.audiofile import read


		def init_processing_chains(cfg: TestConfig) -> None:

ivas_processing_scripts/processing/processing.py

+14 −10

Original line number	Diff line number	Diff line
		@@ -35,14 +35,12 @@ from abc import ABC, abstractmethod
		from itertools import repeat
		from pathlib import Path
		from shutil import copyfile
		from typing import Iterable, Optional, Union
		from typing import Iterable, Union
		from warnings import warn

		import numpy as np

		from ivas_processing_scripts.audiotools import audio
		from ivas_processing_scripts.audiotools.audioarray import window
		from ivas_processing_scripts.audiotools.audiofile import concat, read, trim, write
		from ivas_processing_scripts.audiotools.audiofile import concat, trim
		from ivas_processing_scripts.audiotools.constants import IVAS_FRAME_LEN_MS
		from ivas_processing_scripts.audiotools.convert.__init__ import convert
		from ivas_processing_scripts.audiotools.metadata import (
		@@ -161,7 +159,9 @@ def concat_setup(cfg: TestConfig, chain, logger: logging.Logger):
		logger.info(f"Splits written to file {splits_info_file}")


		def concat_teardown(x, splits, out_fmt, fs, in_fs, meta, len_postamble_ms, logger: logging.Logger):
		def concat_teardown(
		x, splits, out_fmt, fs, in_fs, meta, len_postamble_ms, logger: logging.Logger
		):
		if not splits:
		raise ValueError("Splitting not possible without split marker")

		@@ -182,7 +182,9 @@ def concat_teardown(x, splits, out_fmt, fs, in_fs, meta, len_postamble_ms, logge
		raise ValueError(
		f"Last split index {splits[-1]} is larger than the signal length {len(x)}"
		)
		elif splits[-1] < len(x) - (postamble_len_samples := (len_postamble_ms * fs_old) // 1000):
		elif splits[-1] < len(x) - (
		postamble_len_samples := (len_postamble_ms * fs_old) // 1000
		):
		msg_file_len = len(x)
		if len_postamble_ms > 0:
		msg_file_len = f"(minus postamble length of {postamble_len_samples}): {len(x) - postamble_len_samples}"
		@@ -400,7 +402,9 @@ def process_item(
		copyfile(ppm, out_meta[idx])


		def remove_preamble(x, out_fmt, fs, repeat_signal, preamble_len_ms, postamble_len_ms, meta, logger):
		def remove_preamble(
		x, out_fmt, fs, repeat_signal, preamble_len_ms, postamble_len_ms, meta, logger
		):
		# remove preamble for ISM metadata
		if out_fmt.startswith("ISM"):
		# cut first half of the metadata