Skip to content

Commit 3df2a21

Browse files
authored
add import check (#2076)
Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
1 parent 85dd693 commit 3df2a21

File tree

3 files changed

+18
-0
lines changed

3 files changed

+18
-0
lines changed

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

+6
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
33
test_case="run 3x Torch"
44
echo "${test_case}"
55

6+
echo "##[section]Run import check"
7+
set -e
8+
python -c "import neural_compressor.torch"
9+
python -c "import neural_compressor.common"
10+
echo "##[section]import check pass"
11+
612
# install requirements
713
echo "##[group]set up UT env..."
814
export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH

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

+6
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
33
test_case="run 3x Torch Habana FP8"
44
echo "${test_case}"
55

6+
echo "##[section]Run import check"
7+
set -e
8+
python -c "import neural_compressor.torch"
9+
python -c "import neural_compressor.common"
10+
echo "##[section]import check pass"
11+
612
# install requirements
713
echo "##[group]set up UT env..."
814
export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH

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

+6
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
33
test_case="run 3x TensorFlow"
44
echo "${test_case}"
55

6+
echo "##[section]Run import check"
7+
set -e
8+
python -c "import neural_compressor.tensorflow"
9+
python -c "import neural_compressor.common"
10+
echo "##[section]import check pass"
11+
612
# install requirements
713
echo "##[group]set up UT env..."
814
pip install -r /neural-compressor/test/3x/tensorflow/requirements.txt

0 commit comments

Comments
 (0)