Skip to content

Commit

Permalink
Update setup.py (#11225)
Browse files Browse the repository at this point in the history
removed tf_text requirement from setup.py for tf-models-no-deps pypi package
  • Loading branch information
laxmareddyp authored Jun 18, 2024
1 parent e5c6dba commit 267e679
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions official/pip_package/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,10 @@
from setuptools import find_packages
from setuptools import setup

version = '2.16.0'
tf_version = '2.16.1' # Major version.
version = '2.17.0rc0'
tf_version = '2.17.0rc0' # Major version.

project_name = 'tf-models-official'
project_name = 'tf-models-no-deps'

long_description = """The TensorFlow official models are a collection of
models that use TensorFlow's high-level APIs.
Expand Down Expand Up @@ -67,10 +67,13 @@ 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)
Expand Down

0 comments on commit 267e679

Please sign in to comment.