fix: avoid downloading model for tests

centre-for-humanities-computing · Oct 18, 2023 · 340d6f0 · 340d6f0
1 parent 559fe7d
commit 340d6f0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_relationextraction_component.py b/tests/test_relationextraction_component.py
@@ -1,8 +1,11 @@
+import pytest
+
 from conspiracies.relationextraction import SpacyRelationExtractor  # noqa F401
 
 from .utils import nlp_da  # noqa F401
 
 
+@pytest.mark.skip(reason="Avoid downloading the model on GitHub actions")
 def test_relationextraction_component_pipe(nlp_da):  # noqa F811
     test_sents = [
         "Pernille Blume vinder delt EM-sølv i Ungarn.",
@@ -22,6 +25,7 @@ def test_relationextraction_component_pipe(nlp_da):  # noqa F811
         print(d.text, "\n", d._.relation_triplets)
 
 
+@pytest.mark.skip(reason="Avoid downloading the model on GitHub actions")
 def test_relation_extraction_component_single(nlp_da):  # noqa F811
     nlp_da.add_pipe("relation_extractor", config={"confidence_threshold": 1.8})
     doc = nlp_da("Obama is the former president of the United States.")
@@ -48,12 +52,14 @@ def test_relation_extraction_multi_sentence(nlp_da):  # noqa F811
     ]
 
 
+@pytest.mark.skip(reason="Avoid downloading the model on GitHub actions")
 def test_relation_extraction_empty_string(nlp_da):  # noqa F811
     nlp_da.add_pipe("relation_extractor")
     doc = nlp_da("")
     assert doc._.relation_triplets == []
 
 
+@pytest.mark.skip(reason="Avoid downloading the model on GitHub actions")
 def test_relation_extraction_no_extracted_relation(nlp_da):  # noqa F811
     nlp_da.add_pipe("relation_extractor")
     doc = nlp_da("Ingen relation")