diff --git a/pyproject.toml b/pyproject.toml
index 69a7ada881..d37b0bfd22 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -41,7 +41,7 @@ huggingface = [
     "packaging",
     "tyro",
     "transformers==4.51.3,!=4.47.0",
-    "datasets>=2.16.0",
+    "datasets>=3.4.1",
     "sentencepiece>=0.2.0",
     "tqdm",
     "psutil",
@@ -385,7 +385,7 @@ colab-new = [
     "packaging",
     "tyro",
     "transformers==4.51.3,!=4.47.0",
-    "datasets>=2.16.0",
+    "datasets>=3.4.1",
     "sentencepiece>=0.2.0",
     "tqdm",
     "psutil",