From fdbb0ec2eaa2378ab09a6004a3a7ce9af7eece1b Mon Sep 17 00:00:00 2001 From: struct2tensor-team Date: Fri, 4 Oct 2024 16:16:22 -0700 Subject: [PATCH] updating setup.py to install tf 2.16.2 PiperOrigin-RevId: 682485697 --- setup.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/setup.py b/setup.py index 519b05f..45118d5 100644 --- a/setup.py +++ b/setup.py @@ -67,8 +67,10 @@ def select_constraint(default, nightly=None, git_master=None): setup( name='struct2tensor', version=__version__, - description=('Struct2Tensor is a package for parsing and manipulating' - ' structured data for TensorFlow'), + description=( + 'Struct2Tensor is a package for parsing and manipulating' + ' structured data for TensorFlow' + ), author='Google LLC', author_email='tensorflow-extended-dev@googlegroups.com', # Contained modules and scripts. @@ -79,14 +81,13 @@ def select_constraint(default, nightly=None, git_master=None): 'numpy>=1.22', 'protobuf>=4.25.2,<5;python_version>="3.11"', 'protobuf>=3.20.3,<5;python_version<"3.11"', - 'tensorflow' + select_constraint( - default='>=2.15,<2.16', - nightly='>=2.16.0.dev', - git_master='@git+https://github.com/tensorflow/tensorflow@master'), - 'tensorflow-metadata' + select_constraint( + 'tensorflow>=2.16,<2.17', + 'tensorflow-metadata' + + select_constraint( default='>=1.15.0,<1.16.0', nightly='>=1.16.0.dev', - git_master='@git+https://github.com/tensorflow/metadata@master'), + git_master='@git+https://github.com/tensorflow/metadata@master', + ), 'pyarrow>=10,<11', ], # Add in any packaged data. @@ -119,4 +120,5 @@ def select_constraint(default, nightly=None, git_master=None): keywords='tensorflow protobuf machine learning', long_description=_LONG_DESCRIPTION, long_description_content_type='text/markdown', - cmdclass={'install': _InstallPlatlib}) + cmdclass={'install': _InstallPlatlib}, +)