From c3502f38df71b3d020ea6adcd2cac218deb67008 Mon Sep 17 00:00:00 2001 From: William Grant Date: Wed, 29 Nov 2023 18:15:26 -0500 Subject: [PATCH] bugfix: hyperopt code in docs throws keyerror Signed-off-by: William Grant --- docs/source/getting-started/quickstart-2/index.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/getting-started/quickstart-2/index.rst b/docs/source/getting-started/quickstart-2/index.rst index 7bfab41936021..d79f412844836 100644 --- a/docs/source/getting-started/quickstart-2/index.rst +++ b/docs/source/getting-started/quickstart-2/index.rst @@ -167,7 +167,7 @@ Finally, we will run the hyperparameter sweep using Hyperopt, passing in the ``o # Log the best parameters, loss, and model mlflow.log_params(best) mlflow.log_metric("rmse", best_run["loss"]) - mlflow.tensorflow.log_model(best["model"], "model", signature=signature) + mlflow.tensorflow.log_model(best_run["model"], "model", signature=signature) # Print out the best parameters and corresponding loss print(f"Best parameters: {best}")