import sklearn

Author: winskuo-quic

.ci/scripts/test_model.sh

@@ -167,6 +167,7 @@ test_model_with_qnn() {
   elif [[ "${MODEL_NAME}" == "vit" ]]; then
     EXPORT_SCRIPT=torchvision_vit
   elif [[ "${MODEL_NAME}" == "mb" ]]; then
+    pip install sklearn
     EXPORT_SCRIPT=mobilebert_fine_tune
   elif [[ "${MODEL_NAME}" == "w2l" ]]; then
     EXPORT_SCRIPT=wav2letter