diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt
index d1a68aa4f66f..72a8c0242ec9 100644
--- a/python/requirements/ml/requirements_train.txt
+++ b/python/requirements/ml/requirements_train.txt
@@ -7,5 +7,5 @@ tensorboardX==2.4.1
 # `python/ray/train/examples/transformers/transformers_example.py`
 transformers==4.10.0
 accelerate==0.5.1
-datasets==1.14.0
+datasets==2.14.5
 sentencepiece==0.1.96