python-toolbox/marvin_python_toolbox/management/templates/python-engine/project_package/prediction/predictor.py

#!/usr/bin/env python
# coding=utf-8

"""Predictor engine action.

Use this module to add the project main code.
"""

from .._compatibility import six
from .._logging import get_logger

from marvin_python_toolbox.engine_base import EngineBasePrediction

__all__ = ['Predictor']


logger = get_logger('predictor')


class Predictor(EngineBasePrediction):

    def __init__(self, **kwargs):
        super(Predictor, self).__init__(**kwargs)

    def execute(self, input_message, params, **kwargs):
        """
        Return the predicted value in a json parsable object format.
        Use the self.model and self.metrics objects if necessary.
        """
        return {"predicted_value": "mocked_return"}