fix ut (#1307)

intel · Feb 24, 2024 · 9e9e4c7 · 9e9e4c7
1 parent 8569d7b
commit 9e9e4c7
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/.github/workflows/script/unitTest/run_unit_test_neuralchat.sh b/.github/workflows/script/unitTest/run_unit_test_neuralchat.sh
@@ -71,7 +71,11 @@ function pytest() {
     if [ $(grep -c "Segmentation fault" ${ut_log_name}) != 0 ]; then
        $BOLD_RED && echo "Segmentation Fault found in UT, please check the output..." && $RESET
         exit 1
-    fi  
+    fi
+    if [ $(grep -c "ImportError:" ${ut_log_name}) != 0 ]; then
+       $BOLD_RED && echo "ImportError found in UT, please check the output..." && $RESET
+        exit 1
+    fi
     $BOLD_GREEN && echo "UT finished successfully! " && $RESET
 }
 

diff --git a/.github/workflows/script/unitTest/run_unit_test_neuralspeed.sh b/.github/workflows/script/unitTest/run_unit_test_neuralspeed.sh
@@ -22,7 +22,10 @@ function neuralspeed_test() {
         [ $(grep -c "Segmentation fault" ${ut_log_name}) != 0 ] ||
         [ $(grep -c "core dumped" ${ut_log_name}) != 0 ] ||
         [ $(grep -c "==ERROR:" ${ut_log_name}) != 0 ] ||
-        [ $(grep -c "ModuleNotFoundError:" ${ut_log_name}) != 0 ]; then
+        [ $(grep -c "ModuleNotFoundError:" ${ut_log_name}) != 0 ] ||
+        [ $(grep -c "ImportError:" ${ut_log_name}) != 0 ]; then
+        exit 1
+    fi; then
         $BOLD_RED && echo "Find errors in engine test, please check the output..." && $RESET
         exit 1
     else

diff --git a/.github/workflows/script/unitTest/run_unit_test_optimize.sh b/.github/workflows/script/unitTest/run_unit_test_optimize.sh
@@ -67,7 +67,10 @@ function pytest() {
        $BOLD_RED && echo "Segmentation Fault found in UT, please check the output..." && $RESET
         exit 1
     fi  
-
+    if [ $(grep -c "ImportError:" ${ut_log_name}) != 0 ]; then
+       $BOLD_RED && echo "ImportError found in UT, please check the output..." && $RESET
+        exit 1
+    fi
     $BOLD_GREEN && echo "UT finished successfully! " && $RESET
 }
 

diff --git a/tests/CI/test_llava.py b/tests/CI/test_llava.py
@@ -14,7 +14,7 @@
 
 import os
 import unittest
-from intel_extension_for_transformers.transformers.modeling.llava_models import LlavaMistralForCausalLM
+from intel_extension_for_transformers.transformers.modeling.llava_models.llava_mistral import LlavaMistralForCausalLM
 from accelerate import init_empty_weights
 from transformers import AutoConfig, AutoTokenizer
 import torch