From f5c1c71b4fec8742e7877438f4b2730f6a66e2d3 Mon Sep 17 00:00:00 2001
From: Ubuntu
 <zetia@DevBox-zetia.1jltvvkrfgyuhl3llhmbyldkog.bx.internal.cloudapp.net>
Date: Wed, 6 Dec 2023 10:34:02 +0000
Subject: [PATCH] fix k8s compute update bug

---
 .../kubernetes-online-endpoints-safe-rollout.ipynb   | 12 ++++++++++--
 ...bernetes-online-endpoints-simple-deployment.ipynb | 11 +++++++++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-safe-rollout.ipynb b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-safe-rollout.ipynb
index 31ee4b84fa2..bf0b1464f76 100644
--- a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-safe-rollout.ipynb
+++ b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-safe-rollout.ipynb
@@ -137,9 +137,12 @@
    "source": [
     "from azure.ai.ml import load_compute\n",
     "\n",
+    "\n",
+    "compute_name = \"<COMPUTE_NAME>\"\n",
+    "\n",
     "# for arc connected cluster, the resource_id should be something like '/subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP>/providers/Microsoft.ContainerService/connectedClusters/<CLUSTER_NAME>''\n",
     "compute_params = [\n",
-    "    {\"name\": \"<COMPUTE_NAME>\"},\n",
+    "    {\"name\": compute_name},\n",
     "    {\"type\": \"kubernetes\"},\n",
     "    {\n",
     "        \"resource_id\": \"/subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP>/providers/Microsoft.ContainerService/managedClusters/<CLUSTER_NAME>\"\n",
@@ -147,7 +150,12 @@
     "]\n",
     "k8s_compute = load_compute(source=None, params_override=compute_params)\n",
     "\n",
-    "ml_client.begin_create_or_update(k8s_compute).result()"
+    "compute_list = {c.name:c.type for c in ml_client.compute.list()}\n",
+    "\n",
+    "if compute_name not in compute_list or compute_list[compute_name] != \"kubernetes\":\n",
+    "    ml_client.begin_create_or_update(k8s_compute).result()\n",
+    "else:\n",
+    "    print(\"Compute already exists\")"
    ]
   },
   {
diff --git a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
index 56e0b1041e2..584436224b9 100644
--- a/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
+++ b/sdk/python/endpoints/online/kubernetes/kubernetes-online-endpoints-simple-deployment.ipynb
@@ -321,9 +321,11 @@
    "source": [
     "from azure.ai.ml import load_compute\n",
     "\n",
+    "compute_name = \"<COMPUTE_NAME>\"\n",
+    "\n",
     "# for arc connected cluster, the resource_id should be something like '/subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP>/providers/Microsoft.ContainerService/connectedClusters/<CLUSTER_NAME>''\n",
     "compute_params = [\n",
-    "    {\"name\": \"<COMPUTE_NAME>\"},\n",
+    "    {\"name\": compute_name},\n",
     "    {\"type\": \"kubernetes\"},\n",
     "    {\n",
     "        \"resource_id\": \"/subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP>/providers/Microsoft.ContainerService/managedClusters/<CLUSTER_NAME>\"\n",
@@ -332,7 +334,12 @@
     "k8s_compute = load_compute(source=None, params_override=compute_params)\n",
     "\n",
     "# !!!bug https://msdata.visualstudio.com/Vienna/_workitems/edit/1740311\n",
-    "ml_client.begin_create_or_update(k8s_compute).result()"
+    "compute_list = {c.name:c.type for c in ml_client.compute.list()}\n",
+    "\n",
+    "if compute_name not in compute_list or compute_list[compute_name] != \"kubernetes\":\n",
+    "    ml_client.begin_create_or_update(k8s_compute).result()\n",
+    "else:\n",
+    "    print(\"Compute already exists\")"
    ]
   },
   {