From d03b3467b84086e7951a9ea5f62fd9f68fb59f10 Mon Sep 17 00:00:00 2001
From: Oliver Stanley <olivergestanley@gmail.com>
Date: Thu, 6 Mar 2025 20:07:51 +0000
Subject: [PATCH] Fix constrained generation errors by adding datasets
 dependency (#4142)

---
 python/pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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 = [