From 3fb6e0fbe957157011c78da2351d39021daa9c39 Mon Sep 17 00:00:00 2001
From: Amit Chauhan <70937115+achauhan-scc@users.noreply.github.com>
Date: Fri, 3 Jan 2025 12:32:30 +0530
Subject: [PATCH] Add CLI setup

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

diff --git a/cli/setup.sh b/cli/setup.sh
index 8914f243093..0e23cc7deef 100644
--- a/cli/setup.sh
+++ b/cli/setup.sh
@@ -20,7 +20,8 @@
 
 
 # <az_ml_install>
-az extension add -n ml -y
+# az extension add -n ml -y
+az extension add --source https://azuremlsdktestpypi.blob.core.windows.net/wheels/sdk-cli-v2-public/ml-2.34.0-py3-none-any.whl --yes
 # </az_ml_install>
 
 ## For backward compatibility - running on old subscription