From 684b1ead132f533aef5bfcf6de4466463039de40 Mon Sep 17 00:00:00 2001
From: Chakradhar886 <v-cchakradha@microsoft.com>
Date: Tue, 7 Jul 2026 15:08:03 +0530
Subject: [PATCH 1/2] Update setup.sh

---
 sdk/python/setup.sh | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index ec29a18244..370c77b522 100644
--- a/sdk/python/setup.sh
+++ b/sdk/python/setup.sh
@@ -1,7 +1,6 @@
 #!/bin/bash
 
 # <az_ml_sdk_install>
-# pip install --pre azure-ai-ml
 # </az_ml_sdk_install>
 
 # <mldesigner_install>
@@ -15,12 +14,11 @@ pip install pandas
 
 
 # <az_ml_sdk_test_install>
-# pip install azure-ai-ml==0.1.0.b8
-pip install azure-ai-ml
-# https://docsupport.blob.core.windows.net/ml-sample-submissions/1905732/azure_ai_ml-1.0.0-py3-none-any.whl
+pip install https://azuresdkartifacts.z5.web.core.windows.net/python/distributions/ml-sample/6528728/azure_ai_ml-1.35.0-py3-none-any.whl
 # </az_ml_sdk_test_install>
 
 # protobuf==5.29.0 has IndentationError bug
 pip install "protobuf<=5.28.3"
 
-pip list
\ No newline at end of file
+
+pip list

From 3838af828fecf373cdf7ae2a19cef969f41229f3 Mon Sep 17 00:00:00 2001
From: Chakradhar886 <v-cchakradha@microsoft.com>
Date: Wed, 8 Jul 2026 10:01:49 +0530
Subject: [PATCH 2/2] Update installation command for azure-ai-ml package

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

diff --git a/sdk/python/setup.sh b/sdk/python/setup.sh
index 370c77b522..5b5a73321a 100644
--- a/sdk/python/setup.sh
+++ b/sdk/python/setup.sh
@@ -14,7 +14,7 @@ pip install pandas
 
 
 # <az_ml_sdk_test_install>
-pip install https://azuresdkartifacts.z5.web.core.windows.net/python/distributions/ml-sample/6528728/azure_ai_ml-1.35.0-py3-none-any.whl
+pip install https://azuresdkartifacts.z5.web.core.windows.net/python/distributions/ml-sample/6528728/azure_ai_ml-1.35.0-py3-none-any.whl || pip install azure-ai-ml
 # </az_ml_sdk_test_install>
 
 # protobuf==5.29.0 has IndentationError bug