diff --git a/tutorials/azureml-in-a-day/azureml-in-a-day.ipynb b/tutorials/azureml-in-a-day/azureml-in-a-day.ipynb index 860e7c33d3..e9c62b59d9 100644 --- a/tutorials/azureml-in-a-day/azureml-in-a-day.ipynb +++ b/tutorials/azureml-in-a-day/azureml-in-a-day.ipynb @@ -186,15 +186,12 @@ "dependencies:\n", " - python=3.10\n", " - numpy=1.26.4\n", - " - pip=24.0\n", " - scikit-learn=1.5.1\n", - " - scipy=1.10\n", - " - pandas=1.5.3\n", + " - scipy=1.14.0\n", + " - pandas=2.2.2\n", + " - mlflow=2.14.3\n", " - pip:\n", - " - inference-schema[numpy-support]==1.3.0\n", " - xlrd==2.0.1\n", - " - mlflow\n", - " - azureml-mlflow\n", " - psutil\n", " - tqdm\n", " - ipykernel~=6.29\n", @@ -222,7 +219,7 @@ "source": [ "from azure.ai.ml.entities import Environment\n", "\n", - "custom_env_name = \"aml-scikit-learn\"\n", + "custom_env_name = \"aml-scikit-learn-v2\"\n", "\n", "pipeline_job_env = Environment(\n", " name=custom_env_name,\n", @@ -332,7 +329,7 @@ " mlflow.start_run()\n", "\n", " # enable autologging\n", - " mlflow.sklearn.autolog()\n", + " mlflow.sklearn.autolog(log_models=False)\n", "\n", " ###################\n", " #\n", @@ -453,7 +450,7 @@ " ),\n", " code=\"./src/\", # location of source code\n", " command=\"python main.py --data ${{inputs.data}} --test_train_ratio ${{inputs.test_train_ratio}} --learning_rate ${{inputs.learning_rate}} --registered_model_name ${{inputs.registered_model_name}}\",\n", - " environment=\"aml-scikit-learn@latest\",\n", + " environment=\"aml-scikit-learn-v2@latest\",\n", " experiment_name=\"train_model_credit_default_prediction\",\n", " display_name=\"credit_default_prediction\",\n", ")" @@ -477,7 +474,16 @@ }, "outputs": [], "source": [ - "ml_client.create_or_update(job)" + "returned_job = ml_client.create_or_update(job)" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": {}, + "outputs": [], + "source": [ + "ml_client.jobs.stream(returned_job.name)" ] }, {