From 4e4b988d56880de24dbbb1d3944c6569b5d7e0f2 Mon Sep 17 00:00:00 2001
From: "Sun, Xuehao" <xuehao.sun@intel.com>
Date: Mon, 16 Dec 2024 17:17:19 +0800
Subject: [PATCH] fix

Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
---
 .azure-pipelines/template/ut-template.yml | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/.azure-pipelines/template/ut-template.yml b/.azure-pipelines/template/ut-template.yml
index f31eecbe..54f10966 100644
--- a/.azure-pipelines/template/ut-template.yml
+++ b/.azure-pipelines/template/ut-template.yml
@@ -35,16 +35,14 @@ steps:
   - ${{ if eq(parameters.imageSource, 'build') }}:
     - script: |
         docker exec ${{ parameters.utContainerName }} bash -c "cd /auto-round \
-          && python setup.py bdist_wheel \
-          && pip install $(ls dist/*.whl | head -n 1)[cpu] \
+          && pip install .[cpu] \
           && pip list"
       displayName: "Env Setup"
 
   - ${{ if eq(parameters.imageSource, 'pull') }}:
     - script: |
         docker exec ${{ parameters.utContainerName }} bash -c "cd /auto-round \
-          && python setup.py bdist_wheel \
-          && pip install $(ls dist/*.whl | head -n 1)[hpu] \
+          && pip install .[hpu] \
           && pip list"
       displayName: "HPU Env Setup"