fix: only pass the supported argument to GlobalForecastingModel.predi…

…ct()
unit8co · Nov 3, 2023 · 2f780cf · 2f780cf
1 parent 37fba40
commit 2f780cf
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/darts/utils/historical_forecasts/optimized_historical_forecasts_torch.py b/darts/utils/historical_forecasts/optimized_historical_forecasts_torch.py
@@ -94,12 +94,13 @@ def _optimized_historical_forecasts(
         for cls in model.__class__.__mro__
         if cls.__name__ == "TorchForecastingModel"
     ][0]
+    gfm_kwargs = {
+        k: v
+        for k, v in predict_kwargs.items()
+        if k in ["num_samples", "predict_likelihood_parameters"]
+    }
     super(tfm_cls, model).predict(
-        forecast_horizon,
-        series,
-        past_covariates,
-        future_covariates,
-        **predict_kwargs,
+        forecast_horizon, series, past_covariates, future_covariates, **gfm_kwargs
     )
 
     dataset = model._build_inference_dataset(