code: def load_data(path):
      with gzip.open(find_file(path, "labels.gz")) as flbl:
          struct.unpack(">II", flbl.read(8))
          labels = np.fromstring(flbl.read(), dtype=np.int8)
      with gzip.open(find_file(path, "images.gz")) as fimg:
          _, _, rows, cols = struct.unpack(">IIII", fimg.read(16))
          images = np.fromstring(fimg.read(), dtype=np.uint8).reshape(
              len(labels), rows, cols
          )
          images = (
              images.reshape(images.shape[0], 1, 28, 28).astype(
                  np.float32
              )
              / 255
          )
      return labels, images
  
file_path: sagemaker-python-sdk/tests/data/mxnet_mnist/mnist_neo.py
imports: from __future__ import absolute_import
  import argparse
  import gzip
  import json
  import logging
  import os
  import struct
  import mxnet
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: load_data(path) and returns: labels,  images