fix ut (#1307)

Author: VincyZhang

.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
 }
 

.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

.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
 }
 

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