diff --git a/pyproject.toml b/pyproject.toml
index 7eabe00..c24ef71 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,7 +17,7 @@ dependencies = [
     "ghapi>=1.0.6",
     "gitpython>=3.1.43",
     "pytest>=8.3.3",
-    "datasets==3.0.1",
+    "datasets>=3.0.1,<5.0.0",
     "modal>=0.66.26",
     "strenum>=0.4.15",
     "e2b-code-interpreter>=1.0.4",