add import check (#2076)

Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
intel · Dec 4, 2024 · 3df2a21 · 3df2a21
1 parent 85dd693
commit 3df2a21
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/.azure-pipelines/scripts/ut/3x/run_3x_pt.sh b/.azure-pipelines/scripts/ut/3x/run_3x_pt.sh
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
 test_case="run 3x Torch"
 echo "${test_case}"
 
+echo "##[section]Run import check"
+set -e
+python -c "import neural_compressor.torch"
+python -c "import neural_compressor.common"
+echo "##[section]import check pass"
+
 # install requirements
 echo "##[group]set up UT env..."
 export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH

diff --git a/.azure-pipelines/scripts/ut/3x/run_3x_pt_fp8.sh b/.azure-pipelines/scripts/ut/3x/run_3x_pt_fp8.sh
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
 test_case="run 3x Torch Habana FP8"
 echo "${test_case}"
 
+echo "##[section]Run import check"
+set -e
+python -c "import neural_compressor.torch"
+python -c "import neural_compressor.common"
+echo "##[section]import check pass"
+
 # install requirements
 echo "##[group]set up UT env..."
 export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH

diff --git a/.azure-pipelines/scripts/ut/3x/run_3x_tf.sh b/.azure-pipelines/scripts/ut/3x/run_3x_tf.sh
@@ -3,6 +3,12 @@ python -c "import neural_compressor as nc"
 test_case="run 3x TensorFlow"
 echo "${test_case}"
 
+echo "##[section]Run import check"
+set -e
+python -c "import neural_compressor.tensorflow"
+python -c "import neural_compressor.common"
+echo "##[section]import check pass"
+
 # install requirements
 echo "##[group]set up UT env..."
 pip install -r /neural-compressor/test/3x/tensorflow/requirements.txt