Source code for fortuna.prob_model.calib_config.processor

class CalibProcessor:
    def __init__(
        self,
        devices: int = -1,
        disable_jit: bool = False,
        n_posterior_samples: int = 30,
    ):
        """
        An object to configure computational aspects of the calibration process.

        Parameters
        ----------
        devices: int
            A list of devices to be used during training.
            At the moment two options are supported: use all devices (`devices=-1`) or use no device (`devices=0`).
        disable_jit: bool
            if True, no function within the calibration loop is jitted.
        n_posterior_samples: int
            Number of posterior samples to draw from the posterior distribution for the calibration process.
        """
        self.devices = devices
        self.disable_jit = disable_jit
        self.n_posterior_samples = n_posterior_samples