code: def model_fn(path_to_model_files):
      import neomx  # noqa: F401
  
      ctx = mx.cpu()
      sym, arg_params, aux_params = mx.model.load_checkpoint(
          os.path.join(path_to_model_files, "compiled"), 0
      )
      mod = mx.mod.Module(symbol=sym, context=ctx, label_names=None)
      mod.bind(
          for_training=False,
          data_shapes=[("data", (1, 1, 28, 28))],
          label_shapes=mod._label_shapes,
      )
      mod.set_params(arg_params, aux_params, allow_missing=True)
      return mod
  
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 6 for calling a global function with example usage: model_fn(path_to_model_files) and returns: mod