removed tf_text requirement from setup.py

tensorflow · May 10, 2024 · 9a2b685 · 9a2b685
1 parent a6d3214
commit 9a2b685
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/official/pip_package/setup.py b/official/pip_package/setup.py
@@ -20,10 +20,10 @@
 from setuptools import find_packages
 from setuptools import setup
 
-version = '2.15.0'
-tf_version = '2.15.0'  # Major version.
+version = '2.16.0'
+tf_version = '2.16.1'  # Major version.
 
-project_name = 'tf-models-official'
+project_name = 'tf-models-official-no-deps'
 
 long_description = """The TensorFlow official models are a collection of
 models that use TensorFlow's high-level APIs.
@@ -64,10 +64,14 @@ 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)