fix pylint

Signed-off-by: Sun, Xuehao <[email protected]>
intel · Dec 17, 2024 · 08d4d54 · 08d4d54
1 parent 4e4b988
commit 08d4d54
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.azure-pipelines/scripts/codeScan/pylint/pylint.sh b/.azure-pipelines/scripts/codeScan/pylint/pylint.sh
@@ -15,7 +15,9 @@ RESET="echo -en \\E[0m \\n" # close color
 log_dir="/auto-round/.azure-pipelines/scripts/codeScan/scanLog"
 mkdir -p $log_dir
 
+pip install torch --index-url https://download.pytorch.org/whl/cpu
 pip install -r /auto-round/requirements.txt
+pip install -r /auto-round/requirements-cpu.txt
 
 echo "[DEBUG] list pipdeptree..."
 pip install pipdeptree

diff --git a/.azure-pipelines/template/ut-template.yml b/.azure-pipelines/template/ut-template.yml
@@ -35,6 +35,7 @@ steps:
   - ${{ if eq(parameters.imageSource, 'build') }}:
     - script: |
         docker exec ${{ parameters.utContainerName }} bash -c "cd /auto-round \
+          && pip install torch --index-url https://download.pytorch.org/whl/cpu \
           && pip install .[cpu] \
           && pip list"
       displayName: "Env Setup"