diff --git a/official/pip_package/setup.py b/official/pip_package/setup.py index e912ec47235..ecf2a6d416e 100644 --- a/official/pip_package/setup.py +++ b/official/pip_package/setup.py @@ -20,10 +20,10 @@ from setuptools import find_packages from setuptools import setup -version = '2.15.0' -tf_version = '2.15.0' # Major version. +version = '2.16.0' +tf_version = '2.16.1' # Major version. -project_name = 'tf-models-official' +project_name = 'tf-models-official-no-deps' long_description = """The TensorFlow official models are a collection of models that use TensorFlow's high-level APIs. @@ -64,10 +64,14 @@ def _get_requirements(is_nightly=False): version += '.dev' + datetime.datetime.now().strftime('%Y%m%d') install_requires.append('tf-nightly') install_requires.append('tensorflow-text-nightly') -else: +elif project_name == 'tf-models-official': install_requires, dependency_links = _get_requirements() install_requires.append(f'tensorflow~={tf_version}') install_requires.append(f'tensorflow-text~={tf_version}') +else: + install_requires, dependency_links = _get_requirements() + install_requires.append(f'tensorflow~={tf_version}') + print('install_requires: ', install_requires) print('dependency_links: ', dependency_links)