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

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

diff --git a/.azure-pipelines/template/ut-template.yml b/.azure-pipelines/template/ut-template.yml
index 4efec29b..f31eecbe 100644
--- a/.azure-pipelines/template/ut-template.yml
+++ b/.azure-pipelines/template/ut-template.yml
@@ -43,6 +43,7 @@ steps:
   - ${{ 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 list"
       displayName: "HPU Env Setup"