diff --git a/setup.py b/setup.py index 417e2b4..55ed2b0 100644 --- a/setup.py +++ b/setup.py @@ -49,19 +49,19 @@ def _make_required_install_packages(): "numpy>=1.22.0", 'protobuf>=4.25.2,<6.0.0;python_version>="3.11"', 'protobuf>=4.21.6,<6.0.0;python_version<"3.11"', - "pyarrow>=10,<11", + "pyarrow>=10", "pydot>=1.2,<2", - "tensorflow>=2.17,<2.18", + "tensorflow>=2.16,<2.18", "tensorflow-metadata" + select_constraint( - default=">=1.17.1,<1.18.0", + default=">=1.17.0,<1.18.0", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/metadata@master", ), "tf_keras>=2", "tfx-bsl" + select_constraint( - default=">=1.17.1,<1.18.0", + default=">=1.17.0,<1.18.0", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ),