Update pipeline definition

Author: pdifranc

04-sagemaker-project.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# Step 4: Add a model building CI/CD pipeline\n",
     "\n",
-    "<div class=\"alert alert-warning\"> This notebook has been last tested on a SageMaker Studio JupyterLab instance using the <code>SageMaker Distribution Image 3.6.1</code> and with the SageMaker Python SDK version <code>2.255.0</code></div>\n",
+    "<div class=\"alert alert-warning\"> This notebook has been last tested on a SageMaker Studio JupyterLab instance using the <code>SageMaker Distribution Image 3.7.0</code> and with the SageMaker Python SDK version <code>2.255.0</code></div>\n",
     "\n",
     "In this step you create an automated CI/CD pipeline for model building using [Amazon SageMaker Projects](https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-projects.html). \n",
     "\n",
@@ -1010,15 +1010,16 @@
     "\n",
     "    # If no tracking server ARN, try to find an active MLflow server\n",
     "    if tracking_server_arn is None:\n",
-    "        r = sm.list_mlflow_tracking_servers(\n",
-    "            TrackingServerStatus='Created',\n",
-    "        )['TrackingServerSummaries']\n",
-    "    \n",
-    "        if len(r) < 1:\n",
+    "        summaries = []\n",
+    "        for status in ['Created', 'Updated']:\n",
+    "            r = sm.list_mlflow_apps(Status=status)\n",
+    "            summaries.extend(r['Summaries'])\n",
+    "        \n",
+    "        if len(summaries) < 1:\n",
     "            print(\"You don't have any running MLflow servers. Exiting...\")\n",
     "            return None\n",
     "        else:\n",
-    "            tracking_server_arn = r[0]['TrackingServerArn']\n",
+    "            tracking_server_arn = summaries[0]['Arn']\n",
     "            print(f\"Use the tracking server ARN:{tracking_server_arn}\")\n",
     "        \n",
     "    # Parameters for pipeline execution\n",

05-deploy.ipynb

@@ -9,7 +9,7 @@
    "source": [
     "# Step 5: Add a deployment pipeline\n",
     "\n",
-    "<div class=\"alert alert-warning\"> This notebook has been last tested on a SageMaker Studio JupyterLab instance using the <code>SageMaker Distribution Image 3.6.1</code> and with the SageMaker Python SDK version <code>2.255.0</code></div>\n",
+    "<div class=\"alert alert-warning\"> This notebook has been last tested on a SageMaker Studio JupyterLab instance using the <code>SageMaker Distribution Image 3.7.0</code> and with the SageMaker Python SDK version <code>2.255.0</code></div>\n",
     "\n",
     "In previous step you implemented an automated data processing and model building pipeline. Each run of the pipeline produces a new version of the model. This notebook implements the automated model deployment step in our ML workflow.\n",
     "\n",