From a07485f497b72669f1058639fa806b6127c4c6a9 Mon Sep 17 00:00:00 2001
From: John Aslanides <jaslanides@google.com>
Date: Thu, 18 Feb 2021 02:29:34 -0800
Subject: [PATCH] Relax versioning constraints for tf/tfp, bump patch version
 to 0.3.5.

PiperOrigin-RevId: 358134728
Change-Id: Iba086222e5c97d8e6410097b38668a3935c923dc
---
 bsuite/_metadata.py | 2 +-
 setup.py            | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/bsuite/_metadata.py b/bsuite/_metadata.py
index 598cbccb..f875b3b1 100644
--- a/bsuite/_metadata.py
+++ b/bsuite/_metadata.py
@@ -20,4 +20,4 @@
 a time when bsuite's dependencies may not have been installed yet.
 """
 
-__version__ = '0.3.4'
+__version__ = '0.3.5'
diff --git a/setup.py b/setup.py
index d4b203c2..e317829a 100755
--- a/setup.py
+++ b/setup.py
@@ -25,8 +25,8 @@
 baselines_require = [
     'dm-sonnet',
     'dm-tree',
-    'tensorflow == 2.1',
-    'tensorflow_probability >= 0.8, < 0.9',
+    'tensorflow',
+    'tensorflow_probability',
     'trfl',
     'tqdm',
 ]