Only pin to 2.3 when also installing ScaNN.

PiperOrigin-RevId: 348663892
tensorflow · Dec 22, 2020 · b5af9f4 · b5af9f4
1 parent 673a24e
commit b5af9f4
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,13 @@
 # Changelog
 
+## [0.3.2][2020-12-22]
+
+### Changed
+
+-   Pinned TensorFlow to >= 2.3 when ScaNN is not being installed. When ScaNN is
+    being installed, we pin on >= 2.3, < 2.4. This allows users to use TFRS on
+    TF 2.4 when they are not using ScaNN.
+
 ## [0.3.1][2020-12-22]
 
 ### Changed

diff --git a/setup.py b/setup.py
@@ -17,11 +17,11 @@
 import pathlib
 import setuptools
 
-VERSION = "0.3.1"
+VERSION = "0.3.2"
 
 REQUIRED_PACKAGES = [
     "absl-py >= 0.1.6",
-    "tensorflow == 2.3, < 2.4",
+    "tensorflow >= 2.3",
 ]
 
 long_description = (pathlib.Path(__file__).parent
@@ -40,7 +40,11 @@
     packages=setuptools.find_packages(),
     install_requires=REQUIRED_PACKAGES,
     extras_require={
-        "docs": ["fire", "annoy", "scann == 1.1.1"],
+        "docs": [
+            "fire",
+            "annoy",
+            "scann == 1.1.1",
+            "tensorflow == 2.3, < 2.4",],
     },
     # PyPI package information.
     classifiers=[

diff --git a/tensorflow_recommenders/__init__.py b/tensorflow_recommenders/__init__.py
@@ -21,7 +21,7 @@
 you the flexibility to build complex models.
 """
 
-__version__ = "v0.3.1"
+__version__ = "v0.3.2"
 
 from tensorflow_recommenders import examples
 # Internal extension library import.