Skip to content

Commit 8280799

Browse files
CrickWutensorflower-gardener
authored andcommitted
Separate tensorflow version requirement setup.
PiperOrigin-RevId: 416917874
1 parent 9d6b1c1 commit 8280799

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

official/pip_package/setup.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from setuptools import setup
2222

2323
version = '2.7.0'
24+
tf_version = '2.7.0' # Major version.
2425

2526
project_name = 'tf-models-official'
2627

@@ -61,8 +62,8 @@ def _get_requirements():
6162
install_requires.append('tf-nightly')
6263
install_requires.append('tensorflow-text-nightly')
6364
else:
64-
install_requires.append(f'tensorflow~={version}')
65-
install_requires.append(f'tensorflow-text~={version}')
65+
install_requires.append(f'tensorflow~={tf_version}')
66+
install_requires.append(f'tensorflow-text~={tf_version}')
6667

6768
print('install_requires: ', install_requires)
6869
print('dependency_links: ', dependency_links)

0 commit comments

Comments
 (0)