code: def get_data_module(
      librispeech_path, global_stats_path, sp_model_path
  ):
      train_transform = TrainTransform(
          global_stats_path=global_stats_path,
          sp_model_path=sp_model_path,
      )
      val_transform = ValTransform(
          global_stats_path=global_stats_path,
          sp_model_path=sp_model_path,
      )
      test_transform = TestTransform(
          global_stats_path=global_stats_path,
          sp_model_path=sp_model_path,
      )
      return LibriSpeechDataModule(
          librispeech_path=librispeech_path,
          train_transform=train_transform,
          val_transform=val_transform,
          test_transform=test_transform,
      )
  
imports: import json
  import math
  from functools import partial
  from typing import List
  import sentencepiece
  import torch
  import torchaudio
  from data_module import LibriSpeechDataModule
  from lightning import Batch
label_desc: Use image node_id 4 for calling a global function with example usage: get_data_module(librispeech_path, global_stats_path, sp_model_path) and returns: LibriSpeechDataModule