From 8d20ae7c3bb9e50e0deb04b22bff26c73fd676ef Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Fri, 11 Feb 2022 09:45:33 -0800
Subject: [PATCH] Update the direct `tensorflow-datasets` dependency to
 `~=4.5.2`.

This means that we would accept a hot fix, but not accept major or minor version changes without releasing TFP.

PiperOrigin-RevId: 428023939
---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 31929da..1694e6f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ numpy~=1.21.5
 pandas~=1.1.4
 scipy~=1.2.1
 sklearn~=1.0.2
-tensorflow-datasets>=4.4.0
+tensorflow-datasets~=4.5.2
 tensorflow-estimator>=2.3.0
 tensorflow-probability>=0.13.0
 tensorflow>=1.14
diff --git a/setup.py b/setup.py
index 393b85f..0f1880b 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@ setup(
         'pandas~=1.1.4',
         'scipy~=1.2.1',
         'sklearn~=1.0.2',
-        'tensorflow-datasets>=4.4.0',
+        'tensorflow-datasets~=4.5.2',
         'tensorflow-estimator>=2.3.0',
         'tensorflow-probability>=0.13.0',
     ],