diff --git a/tests/ci/sentence_transformers.sh b/tests/ci/sentence_transformers.sh index 03b3d768a9..e731f9b291 100644 --- a/tests/ci/sentence_transformers.sh +++ b/tests/ci/sentence_transformers.sh @@ -7,6 +7,6 @@ python -m pip install --upgrade pip python -m pip install $OPTIMUM_HABANA_PATH[tests] cd $SENTENCE_TRANSFORMER_PATH/tests python -m pip install .. -pytest test_cmnrl.py test_evaluator.py test_multi_process.py test_train_stsb.py test_compute_embeddings.py test_model_card_data.py test_trainer.py test_util.py test_pretrained_stsb.py +pytest test_cmnrl.py test_multi_process.py test_compute_embeddings.py test_model_card_data.py test_util.py cd $OPTIMUM_HABANA_PATH/tests python -m pytest test_sentence_transformers.py