code: def read_audio(reader, key, audio_format="sound"):
      if audio_format == "sound":
          return reader[key][1]
      elif audio_format == "kaldi_ark":
          return reader[key]
      else:
          raise ValueError(f"Unknown audio format: {audio_format}")
  
imports: import argparse
  import logging
  import re
  import sys
  from pathlib import Path
  from typing import Dict, List, Union
  import numpy
  import torch
  from mir_eval.separation import bss_eval_sources
  from pystoi import stoi
  from typeguard import check_argument_types
  from espnet2.enh.loss.criterions.time_domain import SISNRLoss
  from espnet2.fileio.datadir_writer import DatadirWriter
  from espnet2.fileio.sound_scp import SoundScpReader
  from espnet2.train.dataset import kaldi_loader
  from espnet2.utils import config_argparse
  from espnet2.utils.types import str2bool
  from espnet.utils.cli_utils import get_commandline_args
label_desc: Use image node_id 2 for calling a global function with example usage: read_audio(reader, key, audio_format) and returns: reader, reader