From ce81bdd08ce08974a8076cd5d33b8fcd4acbd821 Mon Sep 17 00:00:00 2001
From: Jianjie Liu <jianjl@microsoft.com>
Date: Tue, 19 Oct 2021 14:30:46 +0000
Subject: [PATCH] Set pyspark upper bound to <3.2.0

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 505fc25784..a8909a6404 100644
--- a/setup.py
+++ b/setup.py
@@ -73,7 +73,7 @@
     "spark": [
         "databricks_cli>=0.8.6,<1",
         "pyarrow>=0.12.1,<6.0.0",
-        "pyspark>=2.4.5,<4.0.0",
+        "pyspark>=2.4.5,<3.2.0",
     ],
     "xlearn": [
         "cmake>=3.18.4.post1",