diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt
index 286c4ab5a929..2ab883f2686c 100644
--- a/python/requirements/ml/requirements_train.txt
+++ b/python/requirements/ml/requirements_train.txt
@@ -12,5 +12,5 @@ tensorboardX==2.4.1
 transformers==4.18.0; python_version <= '3.6'
 transformers==4.19.1; python_version > '3.6'
 accelerate==0.5.1
-datasets==2.0.0
+datasets==2.3.2
 sentencepiece==0.1.96