diff --git a/python/pyproject.toml b/python/pyproject.toml
index 2c2192bb1..fc8332f59 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -38,7 +38,8 @@ runtime_common = [
     "xgrammar==0.1.14",
     "ninja",
     "transformers==4.48.3",
-    "llguidance>=0.6.15"
+    "llguidance>=0.6.15",
+    "datasets"
 ]
 
 srt = [