Skip to content

Commit

Permalink
fix: set self._uses_future_covariates to True in FutureCovariatesLoca…
Browse files Browse the repository at this point in the history
…lForecastingModel when fitted with a future covariate series
  • Loading branch information
madtoinou committed Nov 13, 2023
1 parent ed1e442 commit 37e4afc
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions darts/models/forecasting/forecasting_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -2361,6 +2361,7 @@ def fit(self, series: TimeSeries, future_covariates: Optional[TimeSeries] = None
logger=logger,
)
self._expect_future_covariates = True
self._uses_future_covariates = True

self.encoders = self.initialize_encoders()
if self.encoders.encoding_available:
Expand Down

0 comments on commit 37e4afc

Please sign in to comment.