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',
     ],