def predict_likelihood_parameters(self, model_output: torch.Tensor) -> torch.Tensor:
        """Overwrite parent method since QuantileRegression is not a Likelihood per-se and
        parameters must be extracted differently."""
        num_samples, n_timesteps, n_components, n_quantiles = model_output.shape
        params = model_output.reshape(
            num_samples, n_timesteps, n_components * n_quantiles
        )
        return params
Overwrite parent method since QuantileRegression is not a Likelihood per-se and
        parameters must be extracted differently.
darts/utils/likelihood_models/torch.py