Skip to content

Commit 267e679

Browse files
authoredJun 18, 2024
Update setup.py (#11225)
removed tf_text requirement from setup.py for tf-models-no-deps pypi package
1 parent e5c6dba commit 267e679

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed
 

‎official/pip_package/setup.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
from setuptools import find_packages
2121
from setuptools import setup
2222

23-
version = '2.16.0'
24-
tf_version = '2.16.1' # Major version.
23+
version = '2.17.0rc0'
24+
tf_version = '2.17.0rc0' # Major version.
2525

26-
project_name = 'tf-models-official'
26+
project_name = 'tf-models-no-deps'
2727

2828
long_description = """The TensorFlow official models are a collection of
2929
models that use TensorFlow's high-level APIs.
@@ -67,10 +67,13 @@ def _get_requirements(is_nightly=False):
6767
version += '.dev' + datetime.datetime.now().strftime('%Y%m%d')
6868
install_requires.append('tf-nightly')
6969
install_requires.append('tensorflow-text-nightly')
70-
else:
70+
elif project_name == 'tf-models-official':
7171
install_requires, dependency_links = _get_requirements()
7272
install_requires.append(f'tensorflow~={tf_version}')
7373
install_requires.append(f'tensorflow-text~={tf_version}')
74+
else:
75+
install_requires, dependency_links = _get_requirements()
76+
install_requires.append(f'tensorflow~={tf_version}')
7477

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

0 commit comments

Comments
 (0)