-
Notifications
You must be signed in to change notification settings - Fork 21
/
setup.py
40 lines (36 loc) · 1.51 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from setuptools import setup,find_packages
import sys, os, re
README_FILE = 'README.md'
def get_property(prop, project):
result = re.search(r'{}\s*=\s*[\'"]([^\'"]*)[\'"]'.format(prop), open(project + '/__init__.py').read())
return result.group(1)
project_name = "emlp"
setup(name=project_name,
description="A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groups",
version= get_property('__version__',project_name),
author='Marc Finzi',
author_email='maf820@nyu.edu',
license='MIT',
python_requires='>=3.6',
install_requires=['h5py','objax','pytest','plum-dispatch',
'optax','tqdm>=4.38','matplotlib','scikit-learn'],
extras_require = {
'EXPTS':['olive-oil-ml']
},
packages=find_packages(),
long_description=open('README.md', encoding='UTF-8').read(),
long_description_content_type='text/markdown',
url='https://github.com/mfinzi/equivariant-MLP',
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'Programming Language :: Python :: 3',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
keywords=[
'equivariance','MLP','symmetry','group','AI','neural network',
'representation','group theory','deep learning','machine learning',
'rotation','Lorentz invariance',
],
)