code: def _load_training_data(base_dir):
      x_train = np.load(os.path.join(base_dir, "train_data.npy"))
      y_train = np.load(os.path.join(base_dir, "train_labels.npy"))
      return x_train, y_train
  
file_path: sagemaker-python-sdk/tests/data/tensorflow_mnist/mnist_v1.py
imports: from __future__ import absolute_import, division, print_function
  import argparse
  import json
  import numpy
  import os
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: _load_training_data(base_dir) and returns: x_train,  y_train