diff --git a/pyproject.toml b/pyproject.toml
index b40cfb5af..9db6cd2ff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,6 +25,7 @@ dependencies = [
     "cached_path>=1.6.2",
     "transformers",
     "importlib_resources",
+    "datasets",
 ]
 
 [project.optional-dependencies]