Skip to content

Commit 89321ee

Browse files
committed
fix seed in tests
1 parent 3ad2387 commit 89321ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/pyspark/mllib/tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ def test_classification(self):
310310

311311
rf_model = RandomForest.trainClassifier(
312312
rdd, numClasses=2, categoricalFeaturesInfo=categoricalFeaturesInfo, numTrees=10,
313-
maxBins=4)
313+
maxBins=4, seed=1)
314314
self.assertTrue(rf_model.predict(features[0]) <= 0)
315315
self.assertTrue(rf_model.predict(features[1]) > 0)
316316
self.assertTrue(rf_model.predict(features[2]) <= 0)

0 commit comments

Comments
 (0)