Skip to content

GitLab

Commit b5248395 authored by Yuexin Wu's avatar Yuexin Wu Committed by A. Unique TensorFlower
Browse files

Separate tensorflow version requirement setup. 

PiperOrigin-RevId: 416917874
parent 52bf8afd
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
official/pip_package/setup.py
View file @ b5248395
...@@ -21,6 +21,7 @@ from setuptools import find_packages ...@@ -21,6 +21,7 @@ from setuptools import find_packages
from setuptools import setup from setuptools import setup
version = '2.7.0' version = '2.7.0'
tf_version = '2.7.0' # Major version.
project_name = 'tf-models-official' project_name = 'tf-models-official'
...@@ -61,8 +62,8 @@ if project_name == 'tf-models-nightly': ...@@ -61,8 +62,8 @@ if project_name == 'tf-models-nightly':
install_requires.append('tf-nightly') install_requires.append('tf-nightly')
install_requires.append('tensorflow-text-nightly') install_requires.append('tensorflow-text-nightly')
else: else:
install_requires.append(f'tensorflow~={version}') install_requires.append(f'tensorflow~={tf_version}')
install_requires.append(f'tensorflow-text~={version}') install_requires.append(f'tensorflow-text~={tf_version}')
print('install_requires: ', install_requires) print('install_requires: ', install_requires)
print('dependency_links: ', dependency_links) print('dependency_links: ', dependency_links)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment