diff --git a/aeon/classification/distance_based/tests/test_proximity_forest.py b/aeon/classification/distance_based/tests/test_proximity_forest.py index 9575177327..a107938000 100644 --- a/aeon/classification/distance_based/tests/test_proximity_forest.py +++ b/aeon/classification/distance_based/tests/test_proximity_forest.py @@ -11,7 +11,7 @@ def test_univariate(): """Test that the function gives appropriate error message.""" X, y = load_unit_test() X_multivariate = X.reshape((-1, 2, 12)) - clf = ProximityForest(n_trees=5, random_state=42, n_jobs=-1) + clf = ProximityForest(n_trees=5) with pytest.raises(ValueError): clf.fit(X_multivariate, y) @@ -20,7 +20,7 @@ def test_proximity_forest(): """Test the fit method of ProximityTree.""" X_train, y_train = load_unit_test() X_test, y_test = load_unit_test(split="test") - clf = ProximityForest(n_trees=5, n_splitters=3, max_depth=4) + clf = ProximityForest(n_trees=5, n_splitters=3, max_depth=4, random_state=42) clf.fit(X_train, y_train) y_pred = clf.predict(X_test) score = accuracy_score(y_test, y_pred) diff --git a/pyproject.toml b/pyproject.toml index 768a2d173f..8726769bb0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -64,7 +64,7 @@ all_extras = [ "imbalanced-learn", "keras-self-attention", "kotsu>=0.3.1", - "matplotlib>=3.3.2", + "matplotlib>=3.3.2,<3.9.0", "pmdarima>=1.8.0", "prophet>=1.1.0", "pydot>=2.0.0",