From 4b86ce1afacc6937fda81f6e664e22155e968c7d Mon Sep 17 00:00:00 2001
From: Alexey Volkov <avolkov@google.com>
Date: Tue, 18 Jul 2023 07:02:08 -0700
Subject: [PATCH] fix: Fixed the installation error caused by a PyYAML issue

https://github.com/yaml/pyyaml/issues/601

PiperOrigin-RevId: 548990602
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 93432eec0c..e5a1dae6ab 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,7 @@
     "pyarrow >= 6.0.1",
 ]
 pipelines_extra_require = [
-    "pyyaml>=5.3,<6",
+    "pyyaml>=5.3,<7",
 ]
 datasets_extra_require = [
     "pyarrow >= 3.0.0, < 8.0dev",