Skip to content

Commit dc0c357

Browse files
authored
[MNT] Update tslearn dependency to >=0.7.0 and fix kshape tests (#3119)
* [MNT] Update kshape test expectations for tslearn 0.7.0 * [MNT] fix kshape/ROCKAD tests * [MNT] Fixing tslearn verison requirements * [MNT] Using old version requirements for tslearn
1 parent 7da7726 commit dc0c357

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

aeon/anomaly_detection/series/distance_based/_rockad.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ class ROCKAD(BaseSeriesAnomalyDetector):
8585
>>> detector.fit(X_train)
8686
ROCKAD(...)
8787
>>> detector.predict(X_test)
88-
array([0. , 0.00554713, 0.0699094 , 0.22881059, 0.32382585,
88+
array([0. , 0.00554713, 0.06990941, 0.22881059, 0.32382585,
8989
0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154,
9090
0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154,
9191
0.52382585, 0.65200875, 0.80313368, 0.85194345, 1. ])

aeon/clustering/tests/test_k_shape.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77
from aeon.datasets import load_basic_motions
88
from aeon.utils.validation._dependencies import _check_estimator_deps
99

10-
expected_results = [2, 2, 2, 0, 0]
10+
expected_results = [1, 1, 1, 2, 2]
1111

12-
inertia = 0.5645477840468736
12+
inertia = 0.5753409783429747
1313

14-
expected_iters = 2
14+
expected_iters = 4
1515

16-
expected_labels = [0, 2, 1, 1, 1]
16+
expected_labels = [2, 1, 0, 2, 2]
1717

1818

1919
@pytest.mark.skipif(

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ all_extras = [
7272
"keras>=3.6.0; python_version < '3.13'",
7373
"torch>=1.13.1",
7474
"tsfresh>=0.20.0",
75-
"tslearn>=0.5.2,<0.7.0",
75+
"tslearn>=0.5.2",
7676
]
7777
dl = [
7878
"tensorflow>=2.14; python_version < '3.13'",

0 commit comments

Comments
 (0)