From 0bf6ec7537692a6be4d1bb089f74117285feec11 Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Tue, 14 Nov 2023 00:36:08 -0500
Subject: [PATCH] fix(dependencies): lock build < 1 for now (#643)

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 openllm-python/pyproject.toml | 1 +
 tools/dependencies.py         | 7 ++-----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/openllm-python/pyproject.toml b/openllm-python/pyproject.toml
index 254a90db..656cce58 100644
--- a/openllm-python/pyproject.toml
+++ b/openllm-python/pyproject.toml
@@ -46,6 +46,7 @@ dependencies = [
     "optimum>=1.12.0",
     "accelerate",
     "ghapi",
+    "build[virtualenv]<1",
     "click>=8.1.3",
     "cuda-python;platform_system!=\"Darwin\"",
     "bitsandbytes<0.42",
diff --git a/tools/dependencies.py b/tools/dependencies.py
index f0ed6e09..c3b66f35 100755
--- a/tools/dependencies.py
+++ b/tools/dependencies.py
@@ -7,20 +7,16 @@ import typing as t
 
 import inflection
 import tomlkit
-
 from ghapi.all import GhApi
 
-
 if t.TYPE_CHECKING:
-  from tomlkit.items import Array
-  from tomlkit.items import Table
+  from tomlkit.items import Array, Table
 
 ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.insert(0, os.path.join(ROOT, 'openllm-python', 'src'))
 
 import openllm
 
-
 _OWNER, _REPO = 'bentoml', 'openllm'
 
 
@@ -157,6 +153,7 @@ _BASE_DEPENDENCIES = [
   Dependencies(name='optimum', lower_constraint='1.12.0'),
   Dependencies(name='accelerate'),
   Dependencies(name='ghapi'),
+  Dependencies(name='build', upper_constraint='1', extensions=['virtualenv']),
   Dependencies(name='click', lower_constraint='8.1.3'),
   Dependencies(name='cuda-python', platform=('Darwin', 'ne')),
   Dependencies(name='bitsandbytes', upper_constraint='0.42'),  # 0.41 works with CUDA 11.8