|
1 | | - |
2 | | -from setuptools import setup, find_packages |
| 1 | +from setuptools import find_packages, setup |
3 | 2 |
|
4 | 3 | about = {} |
5 | 4 | with open("e2cnn/__about__.py") as fp: |
6 | 5 | exec(fp.read(), about) |
7 | 6 |
|
8 | 7 | install_requires = [ |
9 | | - 'torch', |
10 | | - 'numpy', |
11 | | - 'scipy', |
12 | | - 'sympy', |
| 8 | + "torch", |
| 9 | + "numpy<1.24", |
| 10 | + "scipy", |
| 11 | + "sympy", |
13 | 12 | ] |
14 | 13 |
|
15 | 14 |
|
16 | 15 | setup_requires = [] |
17 | | -tests_require = ['scikit-learn', 'scikit-image'] |
| 16 | +tests_require = ["scikit-learn", "scikit-image"] |
18 | 17 | extras_require = { |
19 | 18 | # 'RBF-FD and Gaussians': ['cython', 'RBF @ git+https://github.com/treverhines/RBF'] |
20 | 19 | } |
21 | 20 |
|
22 | | -with open("README.md", 'r', encoding='utf-8') as f: |
| 21 | +with open("README.md", "r", encoding="utf-8") as f: |
23 | 22 | long_description = f.read() |
24 | 23 |
|
25 | | -download_url = 'https://github.com/QUVA-Lab/e2cnn/archive/v{}.tar.gz'.format(about['__version__']) |
| 24 | +download_url = "https://github.com/QUVA-Lab/e2cnn/archive/v{}.tar.gz".format( |
| 25 | + about["__version__"] |
| 26 | +) |
26 | 27 |
|
27 | 28 | setup( |
28 | | - name=about['__title__'], |
29 | | - version=about['__version__'], |
30 | | - description=about['__summary__'], |
31 | | - author=about['__author__'], |
32 | | - author_email=about['__email__'], |
33 | | - url=about['__url__'], |
| 29 | + name=about["__title__"], |
| 30 | + version=about["__version__"], |
| 31 | + description=about["__summary__"], |
| 32 | + author=about["__author__"], |
| 33 | + author_email=about["__email__"], |
| 34 | + url=about["__url__"], |
34 | 35 | download_url=download_url, |
35 | | - license=about['__license__'], |
| 36 | + license=about["__license__"], |
36 | 37 | long_description=long_description, |
37 | 38 | long_description_content_type="text/markdown", |
38 | | - packages=find_packages(exclude=['test', 'test.*']), |
39 | | - python_requires='>=3.7', |
| 39 | + packages=find_packages(exclude=["test", "test.*"]), |
| 40 | + python_requires=">=3.7", |
40 | 41 | keywords=[ |
41 | | - 'pytorch', |
42 | | - 'cnn', |
43 | | - 'convolutional-networks' |
44 | | - 'equivariant', |
45 | | - 'isometries', |
| 42 | + "pytorch", |
| 43 | + "cnn", |
| 44 | + "convolutional-networks" "equivariant", |
| 45 | + "isometries", |
46 | 46 | ], |
47 | 47 | install_requires=install_requires, |
48 | 48 | setup_requires=setup_requires, |
|
0 commit comments