Commit 6e01e1cd authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Add tf-text package in the "install_requires" in setup.py.

Our nightly package will depend on tf-text-nighly and stable release will depend on tf-text stable release.

PiperOrigin-RevId: 351216197
parent 0c53959e
...@@ -59,8 +59,10 @@ install_requires, dependency_links = _get_requirements() ...@@ -59,8 +59,10 @@ install_requires, dependency_links = _get_requirements()
if project_name == 'tf-models-nightly': if project_name == 'tf-models-nightly':
version += '.dev' + datetime.datetime.now().strftime('%Y%m%d') version += '.dev' + datetime.datetime.now().strftime('%Y%m%d')
install_requires.append('tf-nightly') install_requires.append('tf-nightly')
install_requires.append('tensorflow-text-nightly')
else: else:
install_requires.append('tensorflow>=2.4.0') install_requires.append('tensorflow>=2.4.0')
install_requires.append('tensorflow-text>=2.4.0')
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