From 5d2fa2a8a92049f869767753767393404c11480e Mon Sep 17 00:00:00 2001 From: Rahul Kumar <74648335+iamrk04@users.noreply.github.com> Date: Tue, 8 Oct 2024 21:44:40 +0530 Subject: [PATCH] Fix classification and tf nb (#3410) * update tf-mnist nb * fix classification nb * remove unwanted change * fix code style --- .../automl-classification-task-bankmarketing.ipynb | 3 ++- .../jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb b/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb index ac475cfe9b9..f16a09d578e 100644 --- a/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb +++ b/sdk/python/jobs/automl-standalone-jobs/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb @@ -954,7 +954,8 @@ "with open(model_file, \"r\") as model_stream:\n", " model_yaml = yaml.safe_load(model_stream)\n", " training_environment_name = (\n", - " \"AzureML-AutoML:\" + model_yaml[\"metadata\"][\"azureml.base_image\"].split(\":\")[-1]\n", + " \"AzureML-ai-ml-automl:\"\n", + " + model_yaml[\"metadata\"][\"azureml.base_image\"].split(\":\")[-1]\n", " )\n", " print(\"Training emvironment {}\".format(training_environment_name))" ] diff --git a/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb b/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb index 855d27b15c1..4c8cf4c12f2 100644 --- a/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb +++ b/sdk/python/jobs/single-step/tensorflow/mnist/tensorflow-mnist.ipynb @@ -128,7 +128,7 @@ "job = command(\n", " code=\"./src\", # local path where the code is stored\n", " command=\"python main.py\",\n", - " environment=\"AzureML-tensorflow-2.12-cuda11@latest\",\n", + " environment=\"AzureML-tensorflow-2.16-cuda11@latest\",\n", " display_name=\"tensorflow-mnist-example\"\n", " # experiment_name: tensorflow-mnist-example\n", " # description: Train a basic neural network with TensorFlow on the MNIST dataset.\n",