def _get_median_prediction(self, model_output: np.ndarray) -> np.ndarray:
        """
        Gets the median prediction per component extracted from the model output.
        """

        Gets the median prediction per component extracted from the model output.
        
darts/utils/likelihood_models/sklearn.py