Update setup.sh

Azure · Jun 11, 2024 · 42c674e · 42c674e
1 parent 0cc1496
commit 42c674e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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>