Fix classification and tf nb (#3410)

* update tf-mnist nb * fix classification nb * remove unwanted change * fix code style
Azure · Oct 8, 2024 · 5d2fa2a · 5d2fa2a
1 parent 1f76114
commit 5d2fa2a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...s/automl-classification-task-bankmarketing/automl-classification-task-bankmarketing.ipynb b/...s/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
@@ -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",