|
1 | 1 | from setuptools import setup, find_packages |
2 | 2 |
|
3 | 3 |
|
4 | | -test_packages = [ |
5 | | - "pytest>=5.4.3", |
6 | | - "pytest-cov>=2.6.1" |
7 | | -] |
| 4 | +test_packages = ["pytest>=5.4.3", "pytest-cov>=2.6.1"] |
8 | 5 |
|
9 | 6 | docs_packages = [ |
10 | 7 | "mkdocs==1.1", |
|
21 | 18 | "matplotlib>= 3.2.2", |
22 | 19 | "seaborn>= 0.11.0", |
23 | 20 | "rapidfuzz>= 0.13.1", |
24 | | - "scikit_learn>= 0.22.2.post1" |
| 21 | + "scikit_learn>= 0.22.2.post1", |
25 | 22 | ] |
26 | 23 |
|
27 | | -gensim_packages = [ |
28 | | - "gensim>=4.0.0" |
29 | | -] |
| 24 | +gensim_packages = ["gensim>=4.0.0"] |
30 | 25 |
|
31 | | -sbert_packages = [ |
32 | | - "sentence-transformers>=0.4.1" |
33 | | -] |
| 26 | +sbert_packages = ["sentence-transformers>=0.4.1"] |
34 | 27 |
|
35 | | -fast_cosine = [ |
36 | | - "sparse_dot_topn>=0.2.9" |
37 | | -] |
| 28 | +fast_cosine = ["sparse_dot_topn>=0.2.9"] |
38 | 29 |
|
39 | | -embeddings_packages = [ |
40 | | - "torch>=1.4.0", |
41 | | - "flair>= 0.7" |
42 | | -] |
| 30 | +embeddings_packages = ["torch>=1.4.0", "flair>= 0.7"] |
43 | 31 |
|
44 | | -spacy_packages = [ |
45 | | - "spacy>=3.0.1" |
46 | | -] |
| 32 | +spacy_packages = ["spacy>=3.0.1"] |
47 | 33 |
|
48 | | -use_packages = [ |
49 | | - "tensorflow", |
50 | | - "tensorflow_hub", |
51 | | - "tensorflow_text" |
52 | | -] |
| 34 | +use_packages = ["tensorflow", "tensorflow_hub", "tensorflow_text"] |
53 | 35 |
|
54 | 36 |
|
55 | | -extra_packages = embeddings_packages + fast_cosine + sbert_packages + spacy_packages + use_packages |
| 37 | +extra_packages = ( |
| 38 | + embeddings_packages + fast_cosine + sbert_packages + spacy_packages + use_packages |
| 39 | +) |
56 | 40 |
|
57 | 41 | dev_packages = docs_packages + test_packages + extra_packages |
58 | 42 |
|
|
62 | 46 | setup( |
63 | 47 | name="polyfuzz", |
64 | 48 | packages=find_packages(exclude=["notebooks", "docs"]), |
65 | | - version="0.4.2", |
| 49 | + version="0.4.3", |
66 | 50 | author="Maarten Grootendorst", |
67 | 51 | author_email="maartengrootendorst@gmail.com", |
68 | 52 | description="PolyFuzz performs fuzzy string matching, grouping, and evaluation.", |
|
102 | 86 | "use": use_packages, |
103 | 87 | "gensim": gensim_packages, |
104 | 88 | }, |
105 | | - python_requires='>=3.9', |
| 89 | + python_requires=">=3.9", |
106 | 90 | ) |
0 commit comments