code: def preprocess_ds_for_fastpitch_align(dataloader):
      pitch_list = []
      for batch in tqdm(dataloader, total=len(dataloader)):
          (
              audios,
              audio_lengths,
              tokens,
              tokens_lengths,
              align_prior_matrices,
              pitches,
              pitches_lengths,
              *_,
          ) = batch
          pitch = pitches.squeeze(0)
          pitch_list.append(pitch[pitch != 0])
  
      get_pitch_stats(pitch_list)
  
imports: import torch
  from hydra.utils import instantiate
  from tqdm import tqdm
  from nemo.core.config import hydra_runner
label_desc: Use image node_id 2 for calling a global function with example usage: preprocess_ds_for_fastpitch_align(dataloader) without return types