fix random state in performance test

wildboar-foundation · Aug 15, 2024 · 56b4cfe · 56b4cfe
1 parent 8c0ead1
commit 56b4cfe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/wildboar/tree/test_shapelet_tree.py b/tests/wildboar/tree/test_shapelet_tree.py
@@ -417,9 +417,9 @@ def test_impurity_equality_tolerance():
 @pytest.mark.parametrize(
     "clf",
     [
-        ShapeletTreeClassifier(strategy="random"),
-        ExtraShapeletTreeClassifier(),
-        ShapeletTreeClassifier(impurity_equality_tolerance=0.0),
+        ShapeletTreeClassifier(strategy="random", random_state=1),
+        ExtraShapeletTreeClassifier(random_state=1),
+        ShapeletTreeClassifier(impurity_equality_tolerance=0.0, random_state=1),
     ],
 )
 @pytest.mark.parametrize("metric", ["euclidean"])