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
41
42
43
44
45
46
47
|
from setuptools import setup
from setuptools import find_packages
from os.path import join, dirname
with open(join(dirname(__file__), "eagerpy/VERSION")) as f:
version = f.read().strip()
try:
# obtain long description from README
readme_path = join(dirname(__file__), "README.rst")
with open(readme_path, encoding="utf-8") as f:
README = f.read()
# remove raw html not supported by PyPI
README = "\n".join(README.split("\n")[3:])
except IOError:
README = ""
install_requires = ["numpy", "typing-extensions>=3.7.4.1"]
tests_require = ["pytest>=5.3.5", "pytest-cov>=2.8.1"]
setup(
name="eagerpy",
version=version,
description="EagerPy is a thin wrapper around PyTorch, TensorFlow Eager, JAX and NumPy that unifies their interface and thus allows writing code that works natively across all of them.",
long_description=README,
long_description_content_type="text/x-rst",
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.6",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
keywords="",
author="Jonas Rauber",
author_email="jonas.rauber@bethgelab.org",
url="https://github.com/jonasrauber/eagerpy",
license="MIT License",
packages=find_packages(),
include_package_data=True,
zip_safe=False,
install_requires=install_requires,
extras_require={"testing": tests_require},
)
|
