From 3ab0e89495696c66a58ce00ba7aaebfdf3d95e62 Mon Sep 17 00:00:00 2001
From: Amit Chauhan <70937115+achauhan-scc@users.noreply.github.com>
Date: Mon, 3 Nov 2025 13:30:03 +0530
Subject: [PATCH 1/2] validate latest CLI wheel 2.40.0

---
 cli/setup.sh | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f24309..b0fa1b4cc6 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -5,22 +5,22 @@
 ###  * Uncomment the following block surrounded by {}
 ###  * Comment the ml extension install within <az_ml_install>
 
-# {
-#      wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.9.0-py3-none-any.whl'
-#
-#      az extension remove -n ml
-#      if ! az extension add --yes --upgrade --source "$wheel_url"; then
-#
-#          echo "Error: Failed to install release candidate"
-#          exit 1
-#      fi
-#      az version
-#      unset wheel_url
-#  }
+{
+      wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.40.0-py3-none-any.whl'
+
+      az extension remove -n ml
+      if ! az extension add --yes --upgrade --source "$wheel_url"; then
+
+          echo "Error: Failed to install release candidate"
+          exit 1
+      fi
+      az version
+      unset wheel_url
+  }
 
 
 # <az_ml_install>
-az extension add -n ml -y
+#az extension add -n ml -y
 # </az_ml_install>
 
 ## For backward compatibility - running on old subscription
@@ -44,4 +44,4 @@ then
     echo "Default resource group set to $GROUP"
 else
     echo "Workflows are using the new subscription."
-fi
\ No newline at end of file
+fi

From 9bd0201ce3a408fa174300ff7069286c3ce1ae9d Mon Sep 17 00:00:00 2001
From: Amit Chauhan <70937115+achauhan-scc@users.noreply.github.com>
Date: Mon, 3 Nov 2025 15:44:28 +0530
Subject: [PATCH 2/2] providing correct URL

---
 cli/setup.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cli/setup.sh b/cli/setup.sh
index b0fa1b4cc6..46620e6945 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -6,7 +6,7 @@
 ###  * Comment the ml extension install within <az_ml_install>
 
 {
-      wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.40.0-py3-none-any.whl'
+      wheel_url='https://mlsdkfdnprod.azureedge.net/ml-2.40.0-py3-none-any.whl'
 
       az extension remove -n ml
       if ! az extension add --yes --upgrade --source "$wheel_url"; then