From a847fa364303ad5883c42d3c06b5d091d62e2565 Mon Sep 17 00:00:00 2001
From: Chen Shen <scv119@gmail.com>
Date: Mon, 14 Feb 2022 08:06:11 -0800
Subject: [PATCH] [Dataset] avoid pyarrow 7.0.0 for dataset (#22253) (#22330)

---
 python/requirements.txt | 2 +-
 python/setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/requirements.txt b/python/requirements.txt
index dc329dd62..5b322fa1d 100644
--- a/python/requirements.txt
+++ b/python/requirements.txt
@@ -46,7 +46,7 @@ starlette
 aiorwlock
 
 # Requirements for running tests
-pyarrow==4.0.1
+pyarrow >= 4.0.1, < 7.0.0
 blist; platform_system != "Windows"
 azure-cli-core==2.29.1
 azure-identity==1.7.0
diff --git a/python/setup.py b/python/setup.py
index debf1fcc4..e26d55f42 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -194,7 +194,7 @@ if setup_spec.type == SetupType.RAY:
     setup_spec.extras = {
         "data": [
             "pandas",
-            "pyarrow>=4.0.1",
+            "pyarrow >= 4.0.1, < 7.0.0",
             "fsspec",
         ],
         "default": [