code: def train_dataloader_creator(config):
      return DataLoader(
          CustomDataset(size=1000),
          batch_size=config["batch_size"],
          shuffle=config["shuffle"],
      )
  
imports: import os
  from unittest import TestCase
  import numpy
  import pytest
  import torch
  import torch.nn
  from torch.utils.data import Dataset, DataLoader
  from zoo.orca.automl.auto_estimator import AutoEstimator
  from zoo.orca.automl.pytorch_utils import LR_NAME
  from zoo.orca.automl import hp
label_desc: Use image node_id 1 for calling a global function with example usage: train_dataloader_creator(config) and returns: DataLoader