diff --git a/aeon/anomaly_detection/series/distance_based/_rockad.py b/aeon/anomaly_detection/series/distance_based/_rockad.py index c83072a7f8..62a90b5183 100644 --- a/aeon/anomaly_detection/series/distance_based/_rockad.py +++ b/aeon/anomaly_detection/series/distance_based/_rockad.py @@ -85,7 +85,7 @@ class ROCKAD(BaseSeriesAnomalyDetector): >>> detector.fit(X_train) ROCKAD(...) >>> detector.predict(X_test) - array([0. , 0.00554713, 0.0699094 , 0.22881059, 0.32382585, + array([0. , 0.00554713, 0.06990941, 0.22881059, 0.32382585, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.43652154, 0.52382585, 0.65200875, 0.80313368, 0.85194345, 1. ]) diff --git a/aeon/clustering/tests/test_k_shape.py b/aeon/clustering/tests/test_k_shape.py index 8af9743004..32c11beba8 100644 --- a/aeon/clustering/tests/test_k_shape.py +++ b/aeon/clustering/tests/test_k_shape.py @@ -7,13 +7,13 @@ from aeon.datasets import load_basic_motions from aeon.utils.validation._dependencies import _check_estimator_deps -expected_results = [2, 2, 2, 0, 0] +expected_results = [1, 1, 1, 2, 2] -inertia = 0.5645477840468736 +inertia = 0.5753409783429747 -expected_iters = 2 +expected_iters = 4 -expected_labels = [0, 2, 1, 1, 1] +expected_labels = [2, 1, 0, 2, 2] @pytest.mark.skipif( diff --git a/pyproject.toml b/pyproject.toml index 98102b9bc9..7a0fa0ae44 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -72,7 +72,7 @@ all_extras = [ "keras>=3.6.0; python_version < '3.13'", "torch>=1.13.1", "tsfresh>=0.20.0", - "tslearn>=0.5.2,<0.7.0", + "tslearn>=0.5.2", ] dl = [ "tensorflow>=2.14; python_version < '3.13'",