Skip to content

Commit

Permalink
fix: updated docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
madtoinou committed Nov 3, 2023
1 parent e7ae306 commit 37fba40
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion darts/models/forecasting/forecasting_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -1452,6 +1452,9 @@ def gridsearch(
must be between `0` and the total number of parameter combinations.
If a float, `n_random_samples` is the ratio of parameter combinations selected from the full grid and must
be between `0` and `1`. Defaults to `None`, for which random selection will be ignored.
fit_kwargs
Additional arguments passed to the model `fit()` method, for example `max_samples_per_ts`,
`n_jobs_multiouput_wrapper` or `num_loader_workers`.
predict_kwargs
Additional arguments passed to the model `predict()` method, for example `predict_likelihood_parameters` or
`num_loader_workers`.
Expand Down Expand Up @@ -2365,7 +2368,6 @@ def _fit_wrapper(
epochs: int = 0,
num_loader_workers: int = 0,
):
"""Propagate the supported parameters to the underlying model fit() method"""
kwargs = dict()
if getattr(self, "trainer_params", False):
kwargs["trainer"] = trainer
Expand Down

0 comments on commit 37fba40

Please sign in to comment.