code: def test_build_train_data_loader(
      self, model_sample, train_dataset, training_config_batch_size
  ):
      trainer = CoupledOptimizerTrainer(
          model=model_sample,
          train_dataset=train_dataset,
          training_config=training_config_batch_size,
      )
  
      train_data_loader = trainer.get_train_dataloader(train_dataset)
  
      assert issubclass(
          type(train_data_loader), torch.utils.data.DataLoader
      )
      assert train_data_loader.dataset == train_dataset
  
      assert (
          train_data_loader.batch_size
          == trainer.training_config.per_device_train_batch_size
      )
  
imports: import os
  from copy import deepcopy
  import pytest
  import torch
  from pythae.models import RAE_L2, RAE_L2_Config
  from pythae.trainers import CoupledOptimizerTrainer, CoupledOptimizerTrainerConfig
  from tests.data.custom_architectures import *
label_desc: Use image node_id 2 for calling the Test_DataLoader obj's underlying member method code with example usage: obj.test_build_train_data_loader(model_sample, train_dataset, training_config_batch_size) without return types