def _set_model_param_names()

in orbit/template/dlt.py [0:0]

• 21 lines of code
• 8 McCabe index (conditional complexity)

    def _set_model_param_names(self):
        """Set posteriors keys to extract from sampling/optimization api
        Notes
        -----
        Overriding :func: `~orbit.models.BaseETS._set_model_param_names`
        It sets additional required attributes related to trend and regression
        """
        self._model_param_names += [param.value for param in BaseSamplingParameters]

        # append seasonality param names
        if self._seasonality > 1:
            self._model_param_names += [
                param.value for param in SeasonalitySamplingParameters
            ]

        # append regressors if any
        if self.num_of_regressors > 0:
            self._model_param_names += [
                RegressionSamplingParameters.REGRESSION_COEFFICIENTS.value
            ]

        if self._global_trend_option != GlobalTrendOption.flat.value:
            self._model_param_names += [
                param.value for param in GlobalTrendSamplingParameters
            ]
        else:
            self._model_param_names += [
                GlobalTrendSamplingParameters.GLOBAL_TREND_LEVEL.value
            ]
            self._model_param_names += [
                GlobalTrendSamplingParameters.GLOBAL_TREND.value
            ]