(
records: List[Dict[str, Any]],
path_to_codes: Dict[str, List[List[int]]],
)
| 127 | |
| 128 | |
| 129 | def attach_reference_audio_codes( |
| 130 | records: List[Dict[str, Any]], |
| 131 | path_to_codes: Dict[str, List[List[int]]], |
| 132 | ) -> None: |
| 133 | for record in records: |
| 134 | ref_audio = normalize_audio_path_list(record.get("ref_audio"), "ref_audio") |
| 135 | if ref_audio is not None: |
| 136 | if len(ref_audio) != 1: |
| 137 | raise ValueError("`ref_audio` only supports a single path.") |
| 138 | record["ref_audio_codes"] = path_to_codes[ref_audio[0]] |
| 139 | |
| 140 | reference_audio = normalize_audio_path_list(record.get("reference_audio"), "reference_audio") |
| 141 | if reference_audio is not None: |
| 142 | record["reference_audio_codes"] = [path_to_codes[path] for path in reference_audio] |
| 143 | |
| 144 | reference = normalize_audio_path_list(record.get("reference"), "reference", allow_none=True) |
| 145 | if reference is not None: |
| 146 | record["reference_audio_codes"] = [ |
| 147 | None if path is None else path_to_codes[path] |
| 148 | for path in reference |
| 149 | ] |
| 150 | |
| 151 | |
| 152 | def main() -> None: |
no test coverage detected