fix num_samles not being passed to optimized hist fc rountine of torc…

…h models (#2060)
unit8co · Nov 8, 2023 · a5a4306 · a5a4306
1 parent af5b141
commit a5a4306
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/darts/utils/historical_forecasts/optimized_historical_forecasts_torch.py b/darts/utils/historical_forecasts/optimized_historical_forecasts_torch.py
@@ -115,6 +115,7 @@ def _optimized_historical_forecasts(
         dataset,
         trainer=None,
         verbose=verbose,
+        num_samples=num_samples,
         predict_likelihood_parameters=predict_likelihood_parameters,
     )