Skip to content

Commit

Permalink
Additional typo
Browse files Browse the repository at this point in the history
Fixed additional typo errors.
  • Loading branch information
cargecla1 committed Oct 12, 2024
1 parent 5c025ed commit 9d74f18
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions neuralforecast/flavor.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,12 +199,12 @@ def log_model(
**kwargs,
):
"""
Log an ``MLForecast`` model as an MLflow artifact for the current run.
Log an ``NeuralForecast`` model as an MLflow artifact for the current run.
Parameters
----------
model : MLForecast
Fitted ``MLForecast`` model object.
model : NeuralForecast
Fitted ``NeuralForecast`` model object.
artifact_path : str
Run-relative artifact path to save the model to.
conda_env : Union[dict, str], optional (default=None)
Expand Down Expand Up @@ -244,7 +244,7 @@ def log_model(
or None to skip waiting.
pip_requirements : Union[Iterable, str], optional (default=None)
Either an iterable of pip requirement strings
(e.g. ["mlforecast", "-r requirements.txt", "-c constraints.txt"]) or the string
(e.g. ["neuralforecast", "-r requirements.txt", "-c constraints.txt"]) or the string
path to a pip requirements file on the local filesystem
(e.g. "requirements.txt")
extra_pip_requirements : Union[Iterable, str], optional (default=None)
Expand Down

0 comments on commit 9d74f18

Please sign in to comment.