code: def _load_wav(path, sample_rate):
      with tf.io.gfile.GFile(path, "rb") as audio_f:
          audio_segment = (
              tfds.core.lazy_imports.pydub.AudioSegment.from_file(
                  audio_f, format="wav"
              )
              .set_channels(1)
              .set_frame_rate(sample_rate)
          )
      audio = np.array(audio_segment.get_array_of_samples()).astype(
          np.float32
      )
      # Convert from int to float representation.
      audio /= 2 ** (8 * audio_segment.sample_width)
      return audio
  
file_path: datasets/tensorflow_datasets/datasets/groove/groove_dataset_builder.py
imports: from __future__ import annotations
  import collections
  import copy
  import csv
  import io
  import os
  from absl import logging
  import numpy
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  import tensorflow_datasets.public_api
label_desc: Use image node_id 1 for calling a global function with example usage: _load_wav(path, sample_rate) and returns: audio