diff --git a/aeon/classification/feature_based/_fresh_prince.py b/aeon/classification/feature_based/_fresh_prince.py index 051cd5b714..daa5c87cd3 100644 --- a/aeon/classification/feature_based/_fresh_prince.py +++ b/aeon/classification/feature_based/_fresh_prince.py @@ -85,6 +85,7 @@ class FreshPRINCEClassifier(BaseClassifier): "capability:multithreading": True, "capability:train_estimate": True, "algorithm_type": "feature", + "non_deterministic": True, "python_dependencies": "tsfresh", } diff --git a/aeon/classification/feature_based/_tsfresh.py b/aeon/classification/feature_based/_tsfresh.py index c9c9575d60..6e919a1c94 100644 --- a/aeon/classification/feature_based/_tsfresh.py +++ b/aeon/classification/feature_based/_tsfresh.py @@ -88,6 +88,7 @@ class TSFreshClassifier(BaseClassifier): "capability:multithreading": True, "capability:unequal_length": True, "algorithm_type": "feature", + "non_deterministic": True, "python_dependencies": "tsfresh", } diff --git a/aeon/clustering/feature_based/_tsfresh.py b/aeon/clustering/feature_based/_tsfresh.py index 6f78f2df78..207bbe12ae 100644 --- a/aeon/clustering/feature_based/_tsfresh.py +++ b/aeon/clustering/feature_based/_tsfresh.py @@ -82,6 +82,7 @@ class TSFreshClusterer(BaseClusterer): "capability:multithreading": True, "capability:unequal_length": True, "algorithm_type": "feature", + "non_deterministic": True, "python_dependencies": "tsfresh", } diff --git a/aeon/regression/feature_based/_fresh_prince.py b/aeon/regression/feature_based/_fresh_prince.py index 9aa0d8a584..116ecac197 100644 --- a/aeon/regression/feature_based/_fresh_prince.py +++ b/aeon/regression/feature_based/_fresh_prince.py @@ -87,6 +87,7 @@ class FreshPRINCERegressor(BaseRegressor): "capability:multithreading": True, "capability:train_estimate": True, "algorithm_type": "feature", + "non_deterministic": True, "python_dependencies": "tsfresh", } diff --git a/aeon/regression/feature_based/_tsfresh.py b/aeon/regression/feature_based/_tsfresh.py index 32408ba0da..08152945d1 100644 --- a/aeon/regression/feature_based/_tsfresh.py +++ b/aeon/regression/feature_based/_tsfresh.py @@ -69,6 +69,7 @@ class TSFreshRegressor(BaseRegressor): "capability:multithreading": True, "capability:unequal_length": True, "algorithm_type": "feature", + "non_deterministic": True, "python_dependencies": "tsfresh", } diff --git a/aeon/testing/testing_config.py b/aeon/testing/testing_config.py index 3a8a1eedc1..d44954747b 100644 --- a/aeon/testing/testing_config.py +++ b/aeon/testing/testing_config.py @@ -65,15 +65,6 @@ # broken by 0.63.0 numba update, see #3307 attempt to fix "HIVECOTEV2": ["check_classifier_against_expected_results"], "TemporalDictionaryEnsemble": ["check_classifier_against_expected_results"], - # multithreading issue, sometimes produces different results between single - # and multithreading - "FreshPRINCEClassifier": ["check_estimator_multithreading"], - "FreshPRINCERegressor": ["check_estimator_multithreading"], - "TSFreshClassifier": ["check_estimator_multithreading"], - "TSFreshRegressor": ["check_estimator_multithreading"], - "TSFreshClusterer": ["check_estimator_multithreading"], - "TSFreshRelevant": ["check_estimator_multithreading"], - "TSFresh": ["check_estimator_multithreading"], } # Exclude specific tests for estimators here only when numba is disabled diff --git a/aeon/transformations/collection/feature_based/_tsfresh.py b/aeon/transformations/collection/feature_based/_tsfresh.py index 44b8284cec..2038b044d0 100644 --- a/aeon/transformations/collection/feature_based/_tsfresh.py +++ b/aeon/transformations/collection/feature_based/_tsfresh.py @@ -44,6 +44,7 @@ class _TSFresh(BaseCollectionTransformer): "capability:multithreading": True, "capability:unequal_length": True, "fit_is_empty": True, + "non_deterministic": True, "python_dependencies": "tsfresh", }