Skip to content

Commit 7073d36

Browse files
committed
✅ Skip LM tests due to disk constrains
Signed-off-by: Marcos Martinez <Marcos.Martinez.Galindo@ibm.com>
1 parent b878e95 commit 7073d36

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

zshot/tests/utils/test_description_enrichment.py

+6
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import pytest
12
import spacy
23

34
from zshot import PipelineConfig
@@ -7,6 +8,7 @@
78
FineTunedLMExtensionStrategy, SummarizationStrategy, ParaphrasingStrategy, EntropyHeuristic
89

910

11+
@pytest.mark.skip(reason="Too expensive to run on every commit")
1012
def test_pretrained_lm_extension_strategy():
1113
description = "The name of a company"
1214
strategy = PreTrainedLMExtensionStrategy()
@@ -19,6 +21,7 @@ def test_pretrained_lm_extension_strategy():
1921
assert len(desc_variations) == 3 and len(set(desc_variations + [description])) == 4
2022

2123

24+
@pytest.mark.skip(reason="Too expensive to run on every commit")
2225
def test_finetuned_lm_extension_strategy():
2326
description = "The name of a company"
2427
strategy = FineTunedLMExtensionStrategy()
@@ -31,6 +34,7 @@ def test_finetuned_lm_extension_strategy():
3134
assert len(desc_variations) == 3 and len(set(desc_variations + [description])) == 4
3235

3336

37+
@pytest.mark.skip(reason="Too expensive to run on every commit")
3438
def test_summarization_strategy():
3539
description = "The name of a company"
3640
strategy = SummarizationStrategy()
@@ -43,6 +47,7 @@ def test_summarization_strategy():
4347
assert len(desc_variations) == 3 and len(set(desc_variations + [description])) == 4
4448

4549

50+
@pytest.mark.skip(reason="Too expensive to run on every commit")
4651
def test_paraphrasing_strategy():
4752
description = "The name of a company"
4853
strategy = ParaphrasingStrategy()
@@ -55,6 +60,7 @@ def test_paraphrasing_strategy():
5560
assert len(desc_variations) == 3 and len(set(desc_variations + [description])) == 4
5661

5762

63+
@pytest.mark.skip(reason="Too expensive to run on every commit")
5864
def test_entropy_heuristic():
5965
def check_is_tuple(x):
6066
return isinstance(x, tuple) and len(x) == 2 and isinstance(x[0], str) and isinstance(x[1], float)

0 commit comments

Comments
 (0)