diff --git a/LICENSE b/LICENSE index 261eeb9e..c9d2380b 100644 --- a/LICENSE +++ b/LICENSE @@ -186,7 +186,7 @@ same "printed page" as the copyright notice for easier identification within third-party archives. - Copyright [yyyy] [name of copyright owner] + Copyright [2022] [Mario Geiger] Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/setup.cfg b/setup.cfg index bdf0bd25..f5b21540 100644 --- a/setup.cfg +++ b/setup.cfg @@ -7,3 +7,27 @@ exclude = .eggs,*.egg,build,dist,docs/_build,notebook [pydocstyle] select = D415, D403 + +[metadata] +name = e3nn_jax +version = attr: e3nn_jax.__version__ +description = Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors. +long_description = file: README.md +long_description_content_type = text/markdown +url = https://e3nn.org +classifiers = + Programming Language :: Python :: 3.7 +license = Apache License 2.0 +license_file = LICENSE + +[options] +packages = find: +python_requires = >=3.7 +install_requires = + jax + dm-haiku + flax + optax + sympy + numpy + tqdm diff --git a/setup.py b/setup.py index f922c277..7f1a1763 100644 --- a/setup.py +++ b/setup.py @@ -1,56 +1,4 @@ -from setuptools import find_packages, setup +from setuptools import setup -import os -import re - - -# Recommendations from https://packaging.python.org/ -here = os.path.abspath(os.path.dirname(__file__)) -with open(os.path.join(here, "README.md"), encoding="utf-8") as f: - long_description = f.read() - - -def read(*parts): - with open(os.path.join(here, *parts), "r") as fp: - return fp.read() - - -def find_version(*file_paths): - version_file = read(*file_paths) - version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", version_file, re.M) - if version_match: - return version_match.group(1) - raise RuntimeError("Unable to find version string.") - - -setup( - name="e3nn_jax", - version=find_version("e3nn_jax", "__init__.py"), - description="Equivariant convolutional neural networks " - "for the group E(3) of 3 dimensional rotations, translations, and mirrors.", - long_description=long_description, - long_description_content_type="text/markdown", - url="https://e3nn.org", - packages=find_packages(exclude=["examples"]), - install_requires=[ - "jax", - "dm-haiku", - "flax", - "optax", - "sympy", - "numpy", - ], - include_package_data=True, - classifiers=[ - "Programming Language :: Python", - "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3 :: Only", - "Programming Language :: Python :: 3.7", - "Programming Language :: Python :: 3.8", - "License :: OSI Approved :: MIT License", - "Operating System :: OS Independent", - ], - python_requires=">=3.7", - license="MIT", - license_files=["LICENSE"], -) +if __name__ == "__main__": + setup()