From b5af9f4ead3d3584bf3434d5a5b4dbf404bcc821 Mon Sep 17 00:00:00 2001
From: Maciej Kula <maciejkula@google.com>
Date: Tue, 22 Dec 2020 11:17:42 -0800
Subject: [PATCH] Only pin to 2.3 when also installing ScaNN.

PiperOrigin-RevId: 348663892
---
 CHANGELOG.md                        |  8 ++++++++
 setup.py                            | 10 +++++++---
 tensorflow_recommenders/__init__.py |  2 +-
 3 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 9c08187..a28f531 100644
--- 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
index 4283da0..d29fd6b 100644
--- 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
index d324962..e01c782 100644
--- 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.