import logging
from typing import Sequence, Tuple
from ..engine import Task
logger = logging.getLogger(__name__)
[docs]class PredictorTask(Task):
"""Perform a single prediction with :class:`pypesto.engine.Task`.
Designed for use with :class:`pypesto.ensemble.Ensemble`.
Attributes
----------
predictor:
The predictor to use.
x:
The parameter vector to compute predictions with.
sensi_orders:
Specifies which sensitivities to compute, e.g. (0,1) -> fval, grad.
mode:
Whether to compute function values or residuals.
id:
The input ID.
"""
[docs] def __init__(
self,
predictor: 'pypesto.predict.Predictor', # noqa: F821
x: Sequence[float],
sensi_orders: Tuple[int, ...],
mode: str,
id: str,
):
super().__init__()
self.predictor = predictor
self.x = x
self.sensi_orders = sensi_orders
self.mode = mode
self.id = id
[docs] def execute(self) -> 'pypesto.predict.PredictionResult': # noqa: F821
"""Execute and return the prediction."""
logger.info(f"Executing task {self.id}.")
prediction = self.predictor(self.x, self.sensi_orders, self.mode)
return prediction