code: def get_data_module(args, sp_model_path, max_frames=1800):
      train_transform = TrainTransform(
          sp_model_path=sp_model_path, args=args
      )
      val_transform = ValTransform(
          sp_model_path=sp_model_path, args=args
      )
      test_transform = TestTransform(
          sp_model_path=sp_model_path, args=args
      )
      return LRS3DataModule(
          args=args,
          train_transform=train_transform,
          val_transform=val_transform,
          test_transform=test_transform,
          max_frames=max_frames,
      )
  
imports: import random
  from typing import List
  import sentencepiece
  import torch
  import torchvision
  from data_module import LRS3DataModule
  from lightning import Batch
  from lightning_av import AVBatch
label_desc: Use image node_id 3 for calling a global function with example usage: get_data_module(args, sp_model_path, max_frames) and returns: LRS3DataModule