Merge pull request #25 from yinxiangshi/main

Update `setup.py`
JieyuZ2 · Apr 21, 2022 · 245cb17 · 245cb17
2 parents 149981c + 19f9a62
commit 245cb17
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 19 deletions.
diff --git a/README.md b/README.md
@@ -14,6 +14,16 @@
 
 
 ## 🔧 New
+**4/20/22**
+1. We have updated the `setup.py` to make installation more flexible.
+
+Please use `pip install ws-benchmark==1.1.2rc0` to install the latest version. We strongly suggest create a new environment to install wrench. We will bring better compatibility in the next stable release.
+ If you have any problems with installation, please let us know.
+
+Known incompatibilities:
+
+`tensorflow==2.8.0`, `albumentations==0.1.12`
+
 **3/18/22**
 1. Wrench is available on [ws-benchmark](https://pypi.org/project/ws-benchmark/) now, using `pip install ws-benchmark` to qucik install.
 

diff --git a/setup.py b/setup.py
@@ -30,34 +30,33 @@
     packages=find_packages(exclude=("test*",)),
     include_package_data=True,
     install_requires=[
-        'cytoolz==0.11.0',
-        'dill==0.3.4',
-        'flyingsquid==0.0.0a0',
+        'cytoolz>=0.11.0',
+        'dill>=0.3.0,<0.4.0',
+        'flyingsquid>=0.0.0a0',
         'future>=0.18.2',
-        'higher>=0.2.1',
-        'typing-extensions>=3.10.0.0, <=3.10.0.2',
-        'torch==1.9.0',
-        'torchvision==0.10.0',
-        'tqdm>=4.62.1',
-        'transformers>=4.6.1,<=4.7.0',
-        'numpy>=1.19.2,<=1.19.5',
-        'snorkel==0.9.7',
-        'seqeval==1.2.2',
-        'scikit-learn==0.24.2',
-        'optuna>=2.7.0,<=2.8.0',
-        'pandas>=1.1.5',
+        'higher>=0.2',
+        'typing-extensions>=3.10.0.0',
+        'torch>=1.2.0,<2.0.0',
+        'torchvision>=0.10.0',
+        'tqdm>=4.33.0,<5.0.0',
+        'transformers>=4.6.1',
+        'numpy>=1.16.5,<=1.22.3',
+        'snorkel>=0.9.7',
+        'seqeval>=1.2.2',
+        'scikit-learn>=0.20.2,<0.25.0',
+        'optuna>=2.7.0,<=2.10.0',
+        'pandas>=1.1.5,<=2.0.0',
         'pillow>=8.3.2',
-        'sentence-transformers==1.1.1',
+        'sentence-transformers>=1.1.1',
         'openml>=0.12.2',
         'cvxpy>=1.1.13,<=1.1.15',
-        'scipy>=1.5.2,<=1.5.4',
+        'scipy>=1.2.0,<2.0.0',
         'faiss-gpu>=1.7.1',
         'numbskull==0.1.1',
         'numba==0.43.0',
         'snorkel-metal>=0.5.0',
-        'spacy>=3.1.2,<=3.1.5',
         'skweak>=0.2.13',
-        'networkx==2.7',
+        'networkx>=2.2,<2.7',
     ],
     python_requires=">=3.6",
     keywords="machine-learning ai weak-supervision",