From 158502c6aa6926751e63f28a53d56cf27dbd43ea Mon Sep 17 00:00:00 2001
From: Yicheng Luo <ethanluoyc@gmail.com>
Date: Thu, 23 Jun 2022 12:18:54 +0100
Subject: [PATCH] BUGFIX: Unpin tensorflow_datasets

The current pinned version does not work for some of the d4rl locomotion v2 datasets due to the wrong specification in metadata types.
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 40fa8f82a7..5ff83fc31e 100755
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@
 tensorflow = [
     'dm-reverb==0.7.2',
     'keras==2.8.0',
-    'tensorflow-datasets==4.5.2',
+    'tensorflow-datasets',
     'tensorflow-estimator==2.8.0',
     'tensorflow==2.8.0',
     'tensorflow_probability==0.15.0',