code: def _get_test_data_loader(training_dir, **kwargs):
      logger.info("Get test data loader")
      return torch.utils.data.DataLoader(
          datasets.MNIST(
              training_dir,
              train=False,
              transform=transforms.Compose(
                  [
                      transforms.ToTensor(),
                      transforms.Normalize((0.1307,), (0.3081,)),
                  ]
              ),
              download=False,  # True sets a dependency on an external site for our canaries.
          ),
          batch_size=1000,
          shuffle=True,
          **kwargs,
      )
  
imports: import argparse
  import json
  import logging
  import os
  import sys
  import torch
  import torch.distributed
  import torch.nn
  import torch.nn.functional
  import torch.optim
  import torch.utils.data
  import torch.utils.data.distributed
  from torchvision import datasets, transforms
label_desc: Use image node_id 2 for calling a global function with example usage: _get_test_data_loader(training_dir) and returns: torch