fix: Fixed the installation error caused by a PyYAML issue

Ark-kun · copybara-github · commit 4b86ce1afacc · 2023-07-18T07:03:50.000-07:00
yaml/pyyaml#601 PiperOrigin-RevId: 548990602
diff --git 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",
