diff --git a/pyproject.toml b/pyproject.toml
index 5487462016..06d0c2c54e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,7 +48,7 @@ optional-dependencies.extra = [
   # quantization:
   "bitsandbytes>=0.45.2,<0.48; sys_platform=='linux' or sys_platform=='win32'",
   # litgpt.evaluate:
-  "datasets>=2.18,<4",
+  "datasets>=2.18,<5",
   # download:
   "huggingface-hub[hf-transfer]>=0.21",
   "litdata==0.2.51",