fix(dependencies): lock build < 1 for now (#643)

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 14, 2023 · 0bf6ec7 · 0bf6ec7
1 parent b30a412
commit 0bf6ec7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git 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
@@ -7,20 +7,16 @@
 
 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 @@ def from_tuple(cls, *decls: t.Any) -> 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