From 7e7bc9d4ea13876b23efefc5d5c3ca580178c36f Mon Sep 17 00:00:00 2001
From: kshitij-microsoft <kchawla@microsoft.com>
Date: Wed, 23 Oct 2024 12:42:50 +0530
Subject: [PATCH] CLI hotfix version 2.31.1

---
 cli/setup.sh        | 5 ++++-
 sdk/python/setup.sh | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f243093..5463c54ce3d 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -20,9 +20,12 @@
 
 
 # <az_ml_install>
-az extension add -n ml -y
+# az extension add -n ml -y
 # </az_ml_install>
 
+# Use a daily build
+az extension add --source https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.31.1-py3-none-any.whl --yes
+
 ## For backward compatibility - running on old subscription
 # <set_variables>
 GROUP="azureml-examples"
diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index 71b17727f51..008cbebce2c 100644
--- a/sdk/python/setup.sh
+++ b/sdk/python/setup.sh
@@ -16,7 +16,7 @@ pip install pandas
 
 # <az_ml_sdk_test_install>
 # pip install azure-ai-ml==0.1.0.b8
-pip install azure-ai-ml
+pip install azure-ai-ml==1.21.1
 # https://docsupport.blob.core.windows.net/ml-sample-submissions/1905732/azure_ai_ml-1.0.0-py3-none-any.whl
 # </az_ml_sdk_test_install>