code: def load_dataset():
      train_x_path = "/tmp/train-images-idx3-ubyte.gz"  # need to change to local disk
      train_y_path = "/tmp/train-labels-idx1-ubyte.gz"  # need to change to local disk
      valid_x_path = "/tmp/t10k-images-idx3-ubyte.gz"  # need to change to local disk
      valid_y_path = "/tmp/t10k-labels-idx1-ubyte.gz"  # need to change to local disk
  
      train_x = read_image_file(
          check_dataset_exist(train_x_path)
      ).astype(np.float32)
      train_y = read_label_file(
          check_dataset_exist(train_y_path)
      ).astype(np.float32)
      valid_x = read_image_file(
          check_dataset_exist(valid_x_path)
      ).astype(np.float32)
      valid_y = read_label_file(
          check_dataset_exist(valid_y_path)
      ).astype(np.float32)
      return train_x, train_y, valid_x, valid_y
  
file_path: singa/examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/src/eva_engine/phase1/algo/singa_ms/cnn_ms/data/mnist.py
imports: import numpy
  import os
  import sys
  import gzip
  import codecs
label_desc: Use image node_id 2 for calling a global function with example usage: load_dataset() and returns: train_x,  train_y,  valid_x,  valid_y