Add **kwargs to FutureCovariatesLocalForecastingModel._fit function s…

…ignature
unit8co · Jul 19, 2024 · bc3b4fd · bc3b4fd
1 parent 616a44e
commit bc3b4fd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/darts/models/forecasting/forecasting_model.py b/darts/models/forecasting/forecasting_model.py
@@ -2956,7 +2956,12 @@ def fit(
         return self._fit(series, future_covariates=future_covariates, **kwargs)
 
     @abstractmethod
-    def _fit(self, series: TimeSeries, future_covariates: Optional[TimeSeries] = None):
+    def _fit(
+        self,
+        series: TimeSeries,
+        future_covariates: Optional[TimeSeries] = None,
+        **kwargs,
+    ):
         """Fits/trains the model on the provided series.
         DualCovariatesModels must implement the fit logic in this method.
         """