mark heavy test as integration test

flairNLP · Feb 2, 2024 · a24000e · a24000e
1 parent 401d5be
commit a24000e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_biomedical_entity_linking.py b/tests/test_biomedical_entity_linking.py
@@ -1,3 +1,5 @@
+import pytest
+
 from flair.data import Sentence
 from flair.models.entity_mention_linking import (
     Ab3PEntityPreprocessor,
@@ -77,6 +79,7 @@ def test_abbrevitation_resolution():
             assert mention == "weaver smith syndrome"
 
 
+@pytest.mark.integration()
 def test_biomedical_entity_linking():
     sentence = Sentence(
         "The mutation in the ABCD1 gene causes X-linked adrenoleukodystrophy, "