From bbab97a6a84bea0014e417d9e73d3e3da39ffc7b Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Thu, 27 Mar 2025 13:24:54 -0700
Subject: [PATCH] add partial_json_parser and einops (#4827)

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

diff --git a/python/pyproject.toml b/python/pyproject.toml
index c757e1297..7c6261983 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -49,6 +49,8 @@ srt = [
     "torch==2.5.1",
     "cuda-python",
     "outlines>=0.0.44,<=0.1.11",
+    "partial_json_parser",
+    "einops",
 ]
 
 # HIP (Heterogeneous-computing Interface for Portability) for AMD