TST add random state

neurodata · Jun 21, 2023 · a31e1de · a31e1de
1 parent f172880
commit a31e1de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sktree/tests/test_honest_forest.py b/sktree/tests/test_honest_forest.py
@@ -112,7 +112,7 @@ def test_max_samples():
     X[:50] *= -1
     y = [0, 1] * 50
     for ms in max_samples_list:
-        uf = HonestForestClassifier(n_estimators=2, max_samples=ms, bootstrap=True)
+        uf = HonestForestClassifier(n_estimators=2, random_state=0, max_samples=ms, bootstrap=True)
         uf = uf.fit(X, y)
         depths.append(uf.estimators_[0].get_depth())