Skip to content

Commit 4ed9982

Browse files
Update test_ts2vec.py
1 parent 1976177 commit 4ed9982

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

aeon/transformations/collection/contrastive_based/tests/test_ts2vec.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,13 @@
44
import pytest
55

66
from aeon.transformations.collection.contrastive_based import TS2Vec
7+
from aeon.utils.validation._dependencies import _check_soft_dependencies
78

89

10+
@pytest.mark.skipif(
11+
not _check_soft_dependencies("torch", severity="none"),
12+
reason="skip test if required soft dependency torch not available",
13+
)
914
@pytest.mark.parametrize("expected_feature_size", [3, 5, 10])
1015
@pytest.mark.parametrize("n_series", [1, 2, 5])
1116
@pytest.mark.parametrize("n_channels", [1, 2, 3])

0 commit comments

Comments
 (0)