Skip to content

Commit

Permalink
Update tracking.py
Browse files Browse the repository at this point in the history
Signed-off-by: Niels Bantilan <[email protected]>
  • Loading branch information
cosmicBboy authored Jan 6, 2023
1 parent d6a5d3c commit e3c49c0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion plugins/flytekit-mlflow/flytekitplugins/mlflow/tracking.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def my_tensorflow_trainer():
When the task is called in a Flyte backend, the decorator starts a new MLFlow run using the Flyte execution name
by default, or a user-provided ``experiment_name`` in the decorator.
:param fn: Function to generate autologs for.
:param framework: The mlflow module to use for autologging
:param experiment_name: The MLFlow experiment name. If not provided, uses the Flyte execution name.
Expand Down

0 comments on commit e3c49c0

Please sign in to comment.