From 3f8a88a6bf2bdea8e46c7bb8530c279b174ea579 Mon Sep 17 00:00:00 2001
From: Jagadeesh Boyidi <156110142+jaga-work@users.noreply.github.com>
Date: Tue, 11 Jun 2024 23:38:06 +0530
Subject: [PATCH] Update setup.sh

---
 cli/setup.sh | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f24309..a0d75476f4 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -19,6 +19,20 @@
 #  }
 
 
+  {
+       wheel_url='https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.27.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_ml_install>
@@ -44,4 +58,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