make test more memory efficient by only loading the smallest model

flairNLP · Feb 6, 2024 · d7fa7dd · d7fa7dd
1 parent 44c1413
commit d7fa7dd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_biomedical_entity_linking.py b/tests/test_biomedical_entity_linking.py
@@ -90,9 +90,8 @@ def test_biomedical_entity_linking():
     tagger = Classifier.load("hunflair")
     tagger.predict(sentence)
 
-    for entity_type in ["disease", "chemical", "gene", "species"]:
-        linker = EntityMentionLinker.load(f"{entity_type}-linker")
-        linker.predict(sentence)
+    linker = EntityMentionLinker.load("disease-linker")
+    linker.predict(sentence)
 
     for span in sentence.get_spans():
         print(span)