code: def collate_fn_wsj0mix_test(
      samples: List[wsj0mix.SampleType], sample_rate
  ):
      max_num_frames = max(s[1].shape[-1] for s in samples)
  
      mixes, srcs, masks = [], [], []
      for sample in samples:
          mix, src, mask = _fix_num_frames(
              sample, max_num_frames, sample_rate, random_start=False
          )
  
          mixes.append(mix)
          srcs.append(src)
          masks.append(mask)
  
      return Batch(
          torch.stack(mixes, 0),
          torch.stack(srcs, 0),
          torch.stack(masks, 0),
      )
  
file_path: audio/examples/source_separation/utils/dataset/utils.py
imports: from collections import namedtuple
  from functools import partial
  from typing import List
  import torch
  from torchaudio.datasets import LibriMix
  from .None import wsj0mix
label_desc: Use image node_id 4 for calling a global function with example usage: collate_fn_wsj0mix_test(samples, sample_rate) and returns: Batch