- numpy version fix.

obss · Oct 8, 2023 · a8272c8 · a8272c8
1 parent bcf9516
commit a8272c8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -44,10 +44,7 @@ def add_pywin(reqs: List[str]) -> None:
     "pytest-timeout>=2.1.0",
 ]
 
-_PRISM_REQUIREMENTS = [
-    "fairseq==0.9.0",
-    "numpy",
-]
+_PRISM_REQUIREMENTS = ["fairseq==0.9.0", "numpy>=1.23,<1.24"]
 
 _METRIC_REQUIREMENTS = [
     "sacrebleu>=2.0.0",
@@ -56,6 +53,7 @@ def add_pywin(reqs: List[str]) -> None:
     "seqeval==1.2.2",
     "sentencepiece>=0.1.99,<0.2",
     "unbabel-comet==2.0.1",
+    "protobuf>=3.20.3,<3.21",
 ]
 
 _METRIC_REQUIREMENTS.extend(_PRISM_REQUIREMENTS)