Fix Numpy Version (#24)

Fixes #21 and #23
MaartenGr · Jun 16, 2021 · 672c90e · 672c90e
1 parent aec80ef
commit 672c90e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/docs/releases.md b/docs/releases.md
@@ -1,3 +1,7 @@
+v0.3.3  
+- Update numpy to "numpy>=1.20.0" to prevent [this](https://github.com/MaartenGr/PolyFuzz/issues/23) and this [issue](https://github.com/MaartenGr/PolyFuzz/issues/21)
+- Update pytorch to "torch>=1.4.0,<1.7.1" to prevent save_state_warning error   
+
 v0.3.2  
 - Fix exploding memory usage when using `top_n`   
 

diff --git a/polyfuzz/__init__.py b/polyfuzz/__init__.py
@@ -1,2 +1,2 @@
 from .polyfuzz import PolyFuzz
-__version__ = "0.3.2"
+__version__ = "0.3.3"
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 ]
 
 base_packages = [
-    "numpy>= 1.18.5,<=1.19.4",
+    "numpy>=1.20.0",
     "scipy>= 1.3.1",
     "pandas>= 0.25.3",
     "tqdm>=4.41.1",
@@ -25,7 +25,7 @@
 ]
 
 fast_cosine = ["sparse_dot_topn>=0.2.9"]
-embeddings_packages = ["torch>=1.2.0", "flair>= 0.7"]
+embeddings_packages = ["torch>=1.4.0,<1.7.1", "flair>= 0.7"]
 
 extra_packages = embeddings_packages + fast_cosine
 
@@ -37,7 +37,7 @@
 setup(
     name="polyfuzz",
     packages=find_packages(exclude=["notebooks", "docs"]),
-    version="0.3.2",
+    version="0.3.3",
     author="Maarten Grootendorst",
     author_email="maartengrootendorst@gmail.com",
     description="PolyFuzz performs fuzzy string matching, grouping, and evaluation.",