diff --git a/pyproject.toml b/pyproject.toml
index 6076feb..c96b451 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -44,7 +44,7 @@ all = [
     "amazon-ads>=0.6.0",
 ]
 train = [
-    "datasets>=2.0",
+    "datasets>=4.8.4",
     "trl>=0.8.0",
     "peft>=0.10.0",
     "deepspeed>=0.14.0",