From 2dde65f8076c7f2275deeff0b3297d878f1520a3 Mon Sep 17 00:00:00 2001
From: Bobby Wang <wbo4958@gmail.com>
Date: Mon, 21 Nov 2022 16:58:00 +0800
Subject: [PATCH] [ci] reduce pyspark test time (#8324)

---
 tests/test_distributed/test_with_spark/test_spark_local.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_distributed/test_with_spark/test_spark_local.py b/tests/test_distributed/test_with_spark/test_spark_local.py
index ee304fe6b..d756b3c5a 100644
--- a/tests/test_distributed/test_with_spark/test_spark_local.py
+++ b/tests/test_distributed/test_with_spark/test_spark_local.py
@@ -697,13 +697,14 @@ class XgboostLocalTest(SparkTestCase):
         self.assert_model_compatible(model.stages[0], tmp_dir)
 
     def test_classifier_with_cross_validator(self):
-        xgb_classifer = SparkXGBClassifier()
+        xgb_classifer = SparkXGBClassifier(n_estimators=1)
         paramMaps = ParamGridBuilder().addGrid(xgb_classifer.max_depth, [1, 2]).build()
         cvBin = CrossValidator(
             estimator=xgb_classifer,
             estimatorParamMaps=paramMaps,
             evaluator=BinaryClassificationEvaluator(),
             seed=1,
+            parallelism=4,
             numFolds=2,
         )
         cvBinModel = cvBin.fit(self.cls_df_train_large)