diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt
index 792f79f66147..41337319b8d3 100644
--- a/python/requirements/ml/requirements_train.txt
+++ b/python/requirements/ml/requirements_train.txt
@@ -9,5 +9,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.2.1
 sentencepiece==0.1.96