Skip to content

Commit 54627ed

Browse files
committed
Fix MLflow logged-models API compat in pipeline tutorial
The curated sklearn-1.5 environment ships MLflow 2.19+ which uses the logged-models API not supported by AzureML tracking server, causing mlflow.sklearn.log_model() and mlflow.register_model() to crash. - autolog(log_models=False) to prevent autolog hitting the API - Replace log_model() with save_model + log_artifacts + MlflowClient registration which use supported endpoints
1 parent 50f168d commit 54627ed

1 file changed

Lines changed: 11 additions & 7 deletions

File tree

tutorials/get-started-notebooks/pipeline.ipynb

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,7 @@
560560
"mlflow.start_run()\n",
561561
"\n",
562562
"# enable autologging\n",
563-
"mlflow.sklearn.autolog()\n",
563+
"mlflow.sklearn.autolog(log_models=False)\n",
564564
"\n",
565565
"os.makedirs(\"./outputs\", exist_ok=True)\n",
566566
"\n",
@@ -609,11 +609,15 @@
609609
"\n",
610610
" # Registering the model to the workspace\n",
611611
" print(\"Registering the model via MLFlow\")\n",
612-
" mlflow.sklearn.log_model(\n",
613-
" sk_model=clf,\n",
614-
" registered_model_name=args.registered_model_name,\n",
615-
" artifact_path=args.registered_model_name,\n",
616-
" )\n",
612+
" mlflow.sklearn.save_model(sk_model=clf, path=\"model\")\n",
613+
" mlflow.log_artifacts(\"model\", artifact_path=args.registered_model_name)\n",
614+
" run_id = mlflow.active_run().info.run_id\n",
615+
" client = mlflow.tracking.MlflowClient()\n",
616+
" try:\n",
617+
" client.create_registered_model(args.registered_model_name)\n",
618+
" except mlflow.exceptions.MlflowException:\n",
619+
" pass\n",
620+
" client.create_model_version(name=args.registered_model_name, source=f\"runs:/{run_id}/{args.registered_model_name}\", run_id=run_id)\n",
617621
"\n",
618622
" # Saving the model to a file\n",
619623
" mlflow.sklearn.save_model(\n",
@@ -946,4 +950,4 @@
946950
},
947951
"nbformat": 4,
948952
"nbformat_minor": 1
949-
}
953+
}

0 commit comments

Comments
 (0)