code: def arithmetic_train_loader(c: ArithmeticAutoregression):
      """
      Training data loader
      """
      return DataLoader(
          ArithmeticDataset(
              c.seq_len, c.max_digits, c.train_sequences_per_epoch
          ),
          batch_size=c.batch_size,
          collate_fn=transpose_batch,
          num_workers=4,
      )
  
imports: import random
  import string
  from typing import List
  import torch
  from labml.logger import Text
  from torch.utils.data import DataLoader, Dataset
  from labml import monit, logger, tracker
  from labml.configs import option
  from labml_nn.experiments.nlp_autoregression import NLPAutoRegressionConfigs, transpose_batch
label_desc: Use image node_id 1 for calling a global function with example usage: arithmetic_train_loader(c) and returns: DataLoader