def default_predict_fn()

in src/sagemaker_mxnet_serving_container/default_inference_handler.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

    def default_predict_fn(self, data, model):
        """Use the model to create a prediction for the data.

        Args:
            data (mxnet.io.NDArrayIter): input data for prediction
            model (mxnet.module.BaseModule): an MXNet Module

        Returns:
            list: the prediction result. This will be either a list of ``mxnet.nd.array`` or
                a list of lists of ``mxnet.nd.array``

        """
        return model.predict(data)