Change tensorflow version requirement setup.

PiperOrigin-RevId: 416912144

Change tensorflow version requirement setup.
PiperOrigin-RevId: 416912144
52bf8afd · Yuexin Wu · A. Unique TensorFlower · 96674ab0 · 52bf8afd
Commit 52bf8afd authored Dec 16, 2021 by Yuexin Wu Committed by A. Unique TensorFlower Dec 16, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/pip_package/setup.py official/pip_package/setup.py +2 -2

No files found.
--- a/official/pip_package/setup.py
+++ b/official/pip_package/setup.py
@@ -61,8 +61,8 @@ if project_name == 'tf-models-nightly':
  install_requires.append('tf-nightly')
  install_requires.append('tensorflow-text-nightly')
 else:
-  install_requires.append('tensorflow>=2.7.0')
-  install_requires.append('tensorflow-text>=2.7.0')
+  install_requires.append(f'tensorflow~={version}')
+  install_requires.append(f'tensorflow-text~={version}')

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