Fix MHC test case

harmonydata · May 11, 2024 · 66c3292 · 66c3292
1 parent 6582ae9
commit 66c3292
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_match_mhc.py b/tests/test_match_mhc.py
@@ -46,7 +46,7 @@
 
 mhc_questions_as_text = ["Have you ever felt annoyed by criticism of your drinking?", "Have you recently"]
 
-model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-mpnet-base-v2')
+model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
 
 mhc_embeddings = model.encode(np.asarray(mhc_questions_as_text))