Skip to content

Commit aca6b24

Browse files
committed
minor fix
Signed-off-by: chensuyue <suyue.chen@intel.com>
1 parent b326e39 commit aca6b24

File tree

3 files changed

+10
-14
lines changed

3 files changed

+10
-14
lines changed

.azure-pipelines/scripts/install_nc.sh

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,14 @@
22

33
echo -e "\n Install Neural Compressor ... "
44
cd /neural-compressor
5-
if [ $1 ]; then
6-
if [[ $1 = *"3x_pt" ]]; then
7-
python -m pip install --no-cache-dir -r requirements_pt.txt
8-
python setup.py pt bdist_wheel
9-
pip install dist/neural_compressor*.whl
10-
elif [[ $1 = *"3x_tf" ]]; then
11-
python -m pip install --no-cache-dir -r requirements_tf.txt
12-
python setup.py tf bdist_wheel
13-
pip install dist/neural_compressor*.whl
14-
fi
5+
if [[ $1 = *"3x_pt" ]]; then
6+
python -m pip install --no-cache-dir -r requirements_pt.txt
7+
python setup.py pt bdist_wheel
8+
pip install dist/neural_compressor*.whl
9+
elif [[ $1 = *"3x_tf" ]]; then
10+
python -m pip install --no-cache-dir -r requirements_tf.txt
11+
python setup.py tf bdist_wheel
12+
pip install dist/neural_compressor*.whl
1513
else
1614
python -m pip install --no-cache-dir -r requirements.txt
1715
python setup.py 2x bdist_wheel

.azure-pipelines/scripts/ut/3x/run_3x_pt.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,12 @@ echo "${test_case}"
55

66
# install requirements
77
echo "set up UT env..."
8-
pip install -r /neural-compressor/requirements_pt.txt
98
pip install transformers
109
pip install coverage
1110
pip install pytest
1211
pip list
1312

14-
export COVERAGE_RCFILE=/neural-compressor/.azure-pipelines/scripts/ut/coverage.3x_pt
13+
export COVERAGE_RCFILE=/neural-compressor/.azure-pipelines/scripts/ut/3x/coverage.3x_pt
1514
lpot_path=$(python -c 'import neural_compressor; import os; print(os.path.dirname(neural_compressor.__file__))')
1615
cd /neural-compressor/test || exit 1
1716
find ./3x/torch/* -name "test*.py" | sed 's,\.\/,coverage run --source='"${lpot_path}"' --append ,g' | sed 's/$/ --verbose/'> run.sh

.azure-pipelines/scripts/ut/3x/run_3x_tf.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,11 @@ echo "${test_case}"
55

66
# install requirements
77
echo "set up UT env..."
8-
pip install -r /neural-compressor/requirements_tf.txt
98
pip install coverage
109
pip install pytest
1110
pip list
1211

13-
export COVERAGE_RCFILE=/neural-compressor/.azure-pipelines/scripts/ut/coverage.3x_tf
12+
export COVERAGE_RCFILE=/neural-compressor/.azure-pipelines/scripts/ut/3x/coverage.3x_tf
1413
lpot_path=$(python -c 'import neural_compressor; import os; print(os.path.dirname(neural_compressor.__file__))')
1514
cd /neural-compressor/test || exit 1
1615
find ./3x/tensorflow/* -name "test*.py" | sed 's,\.\/,coverage run --source='"${lpot_path}"' --append ,g' | sed 's/$/ --verbose/'> run.sh

0 commit comments

Comments
 (0)