Commit 3d03e4d7 authored by Jan Kiene's avatar Jan Kiene
Browse files

fix split idx selection for OMASA

parent 741ca238

tests/conftest.py

+2 −2
Original line number Diff line number Diff line
@@ -1270,14 +1270,14 @@ def get_split_idx(input_file: str, sampling_rate_khz: int) -> Optional[np.ndarra 


    input_file = input_file.lower()

    if "omasa" in input_file:

        format = "_".join(input_file.replace(".wav", "").split("_")[:-1])

        format = "_".join(input_file.split("_")[1:-1])

    elif "osba" in input_file:

        if "foa" in input_file:

            format = "osba_foa"

        else:

            format = "osba_hoa"

    else:

        format = input_file.replace(".wav", "").split("_")[-1].lower()

        format = input_file.split("_")[-1].lower()

    idx = SPLIT_IDX.get(format, np.empty(0)).copy()



    if len(idx) > 0 and sampling_rate_khz != 16: