Update setup.py

McGill-NLP · May 29, 2024 · 5a219c0 · 5a219c0
1 parent 1a64919
commit 5a219c0
Showing 1 changed file with 32 additions and 13 deletions.
diff --git a/setup.py b/setup.py
@@ -18,24 +18,43 @@
     packages=find_packages(include=['instruct_qa*']),
     install_requires=[
         "torch",
-        "transformers==4.29.1",
-        "datasets==2.13.1",
-        "evaluate==0.4.0",
-        "wget==3.2",
-        "rouge_score==0.1.2",
-        "faiss-cpu==1.7.4",
-        "sentence-transformers==2.2.2",
-        "pyserini==0.19.0",
-        "flask==2.2.5",
+        "transformers>=4.29.1",
+        "datasets>=2.13.1",
+        "evaluate>=0.4.0",
+        "wget>=3.2",
+        "rouge_score>=0.1.2",
+        "faiss-cpu>=1.7.4",
+        "sentence-transformers>=2.2.2",
+        "pyserini>=0.19.0",
+        "flask>=2.2.5",
         "gunicorn",
-        "openai==0.27.8",
+        "openai>=0.27.8",
         "nltk",
-        "accelerate==0.20.3",
+        "accelerate>=0.20.3",
         "jsonlines",
-        "protobuf==3.20.*",
+        "protobuf>=3.20.0",
     ],
     extras_require={
-        "evaluation": ["tensorflow==2.9.0", "tensorflow-text==2.9.0", "bert_score==0.3.13", "allennlp==2.10.1", "allennlp-models==2.10.1"],
+        "evaluation": ["tensorflow>=2.9.0", "tensorflow-text>=2.9.0", "bert_score>=0.3.13", "allennlp>=2.10.1", "allennlp-models>=2.10.1"],
+        "evaluation-reproduce": ["tensorflow==2.9.0", "tensorflow-text==2.9.0", "bert_score==0.3.13", "allennlp==2.10.1", "allennlp-models==2.10.1"],
+        "reproduce": [
+            "torch",
+            "transformers==4.29.1",
+            "datasets==2.13.1",
+            "evaluate==0.4.0",
+            "wget==3.2",
+            "rouge_score==0.1.2",
+            "faiss-cpu==1.7.4",
+            "sentence-transformers==2.2.2",
+            "pyserini==0.19.0",
+            "flask==2.2.5",
+            "gunicorn",
+            "openai==0.27.8",
+            "nltk",
+            "accelerate==0.20.3",
+            "jsonlines",
+            "protobuf==3.20.*",
+        ],
     },
     classifiers=[
         "Programming Language :: Python :: 3",