def predict(self, model_output: np.ndarray, num_samples: int) -> np.ndarray:
        """
        Generates sampled or direct likelihood parameter predictions.

        Parameters
        ----------
        model_output
            The output of the StatsForecast model.
        num_samples
            Number of times a prediction is sampled from the likelihood model / distribution.
            If `1` and `predict_likelihood_parameters=False`, returns median / mean predictions.
        """
        if num_samples > 1:
            return self.sample(model_output, num_samples=num_samples)

        # likelihood parameters or median prediction are handled by the `levels` parameter in
        # `StatsForecastModel._predict()`
        return model_output

        Generates sampled or direct likelihood parameter predictions.

        Parameters
        ----------
        model_output
            The output of the StatsForecast model.
        num_samples
            Number of times a prediction is sampled from the likelihood model / distribution.
            If `1` and `predict_likelihood_parameters=False`, returns median / mean predictions.
        
darts/utils/likelihood_models/statsforecast.py