diff --git a/official/requirements.txt b/official/requirements.txt
index 76106e774d2f0b83bc075b7a8b89c5416a459eda..da31fae022bdc41675d5ae55e2b00eb1d731eae6 100644
--- a/official/requirements.txt
+++ b/official/requirements.txt
@@ -9,6 +9,7 @@ psutil>=5.4.3
 py-cpuinfo>=3.3.0
 scipy>=0.19.1
 tensorflow-hub>=0.6.0
+tensorflow_datasets
 typing
 sentencepiece
 Cython