Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
felipeangelimvieira committed Feb 24, 2025
1 parent 67da6c2 commit 0dc7c4c
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions src/prophetverse/sktime/univariate.py
Original file line number Diff line number Diff line change
Expand Up @@ -195,9 +195,7 @@ def _get_fit_data(self, y, X, fh):
"trend_model": self.trend_model_,
"noise_scale": self.noise_scale,
"scale": self._scale,
"exogenous_effects": (
self.non_skipped_exogenous_effect if self._has_exogenous else None
),
"exogenous_effects": self.non_skipped_exogenous_effect,
}

inputs = {
Expand Down Expand Up @@ -237,9 +235,7 @@ def _get_predict_data(

trend_data = self.trend_model_.transform(X=X, fh=fh_as_index)

exogenous_data = (
self._transform_effects(X, fh_as_index) if self._has_exogenous else None
)
exogenous_data = self._transform_effects(X, fh_as_index)

return dict(
y=None,
Expand Down

0 comments on commit 0dc7c4c

Please sign in to comment.