Update notebooks for custom-output-batch and mnist-batch

Author: trimicrosoft

sdk/python/endpoints/batch/deploy-models/custom-outputs-parquet/custom-output-batch.ipynb

@@ -29,7 +29,8 @@
     "* `lightgbm==1.5.2`\n",
     "* `numpy`\n",
     "* `pandas`\n",
-    "* `pyarrow`"
+    "* `pyarrow`\n",
+    "* `azure`"
    ]
   },
   {
@@ -477,7 +478,7 @@
     "environment = Environment(\n",
     "    name=\"batch-mlflow-xgboost\",\n",
     "    conda_file=\"environment/conda.yaml\",\n",
-    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu22.04:latest\",\n",
     ")"
    ]
   },

sdk/python/endpoints/batch/deploy-models/mnist-classifier/mnist-batch.ipynb

@@ -406,7 +406,7 @@
     "env = Environment(\n",
     "    name=\"batch-torch-py38\",\n",
     "    conda_file=\"deployment-torch/environment/conda.yaml\",\n",
-    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu22.04:latest\",\n",
     ")"
    ]
   },
@@ -576,7 +576,7 @@
     "    endpoint_name=endpoint_name,\n",
     "    deployment_name=deployment.name,\n",
     "    input=Input(\n",
-    "        path=\"https://azuremlexampledata.blob.core.windows.net/data/mnist/sample/\",\n",
+    "        path=\"https://azuremldata2.blob.core.windows.net/data/mnist/sample/\",\n",
     "        type=AssetTypes.URI_FOLDER,\n",
     "    ),\n",
     ")"
@@ -770,7 +770,7 @@
     "job = ml_client.batch_endpoints.invoke(\n",
     "    endpoint_name=endpoint_name,\n",
     "    input=Input(\n",
-    "        path=\"https://azuremlexampledata.blob.core.windows.net/data/mnist/sample/\",\n",
+    "        path=\"https://azuremldata2.blob.core.windows.net/data/mnist/sample/\",\n",
     "        type=AssetTypes.URI_FOLDER,\n",
     "    ),\n",
     "    params_override=[\n",
@@ -802,7 +802,7 @@
     "job = ml_client.batch_endpoints.invoke(\n",
     "    endpoint_name=endpoint_name,\n",
     "    input=Input(\n",
-    "        path=\"https://azuremlexampledata.blob.core.windows.net/data/mnist/sample/\"\n",
+    "        path=\"https://azuremldata2.blob.core.windows.net/data/mnist/sample/\"\n",
     "    ),\n",
     "    params_override=[{\"mini_batch_size\": \"20\"}, {\"compute.instance_count\": \"5\"}],\n",
     ")"