in mxnet_managed_spot_training_checkpointing/source_dir/mnist.py [0:0]
def neo_preprocess(payload, content_type):
import logging
import numpy as np
import io
logging.info('Invoking user-defined pre-processing function')
if content_type != 'application/vnd+python.numpy+binary':
raise RuntimeError('Content type must be application/vnd+python.numpy+binary')
f = io.BytesIO(payload)
return np.load(f)