diff --git a/setup.py b/setup.py
index 99726fe..7360770 100644
--- a/setup.py
+++ b/setup.py
@@ -84,7 +84,7 @@ def select_constraint(default, nightly=None, git_master=None):
             default='>=1.15.0,<1.16.0',
             nightly='>=1.16.0.dev',
             git_master='@git+https://github.com/tensorflow/metadata@master'),
-        'pyarrow>=10,<11',
+        'pyarrow>=10,<17',
     ],
     # Add in any packaged data.
     include_package_data=True,