move import of neomxnet into inference functions

Author: Rui Wang Napieralski

tests/data/mxnet_mnist/mnist_neo.py

@@ -19,7 +19,6 @@
 
 import mxnet as mx
 import numpy as np
-import neomxnet  # noqa: F401
 
 
 def load_data(path):
@@ -106,6 +105,8 @@ def train(
 
 
 def model_fn(path_to_model_files):
+    import neomxnet  # noqa: F401
+
     ctx = mx.cpu()
     sym, arg_params, aux_params = mx.model.load_checkpoint(
         os.path.join(path_to_model_files, "compiled"), 0
@@ -119,6 +120,8 @@ def model_fn(path_to_model_files):
 
 
 def transform_fn(mod, payload, input_content_type, requested_output_content_type):
+    import neomxnet  # noqa: F401
+
     if input_content_type != "application/vnd+python.numpy+binary":
         raise RuntimeError("Input content type must be application/vnd+python.numpy+binary")