[pyspark] fix empty data issue when constructing DMatrix (#8245)

Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
2022-09-20 16:43:20 +08:00
parent 70df36c99c
commit 520586ffa7
5 changed files with 86 additions and 7 deletions
--- a/tests/python/test_spark/utils.py
+++ b/tests/python/test_spark/utils.py
@@ -102,7 +102,7 @@ class SparkTestCase(TestSparkContext, TestTempDir, unittest.TestCase):
    def setUpClass(cls):
        cls.setup_env(
            {
-                "spark.master": "local[2]",
+                "spark.master": "local[4]",
                "spark.python.worker.reuse": "false",
                "spark.driver.host": "127.0.0.1",
                "spark.task.maxFailures": "1",