forked from pymc-devs/pymc4
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
71 lines (58 loc) · 2.16 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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import codecs
import re
from pathlib import Path
from setuptools import setup, find_packages
PROJECT_ROOT = Path(__file__).resolve().parent
REQUIREMENTS_FILE = PROJECT_ROOT / "requirements.txt"
REQUIREMENTS_DEV_FILE = PROJECT_ROOT / "requirements-dev.txt"
README_FILE = PROJECT_ROOT / "README.md"
VERSION_FILE = PROJECT_ROOT / "pymc4" / "__init__.py"
NAME = "pymc4"
DESCRIPTION = "A Python probabilistic programming interface to TensorFlow, for Bayesian modelling and machine learning."
AUTHOR = "PyMC Developers"
AUTHOR_EMAIL = "[email protected]"
URL = ("https://github.com/pymc-devs/pymc4",)
LICENSE = "Apache License, Version 2.0"
CLASSIFIERS = [
"Programming Language :: Python",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.5",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"License :: OSI Approved :: Apache Software License",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Mathematics",
"Operating System :: OS Independent",
]
def get_requirements(path):
with codecs.open(path) as buff:
return buff.read().splitlines()
def get_long_description():
with codecs.open(README_FILE, "rt") as buff:
return buff.read()
def get_version():
lines = open(VERSION_FILE, "rt").readlines()
version_regex = r"^__version__ = ['\"]([^'\"]*)['\"]"
for line in lines:
mo = re.search(version_regex, line, re.M)
if mo:
return mo.group(1)
raise RuntimeError("Unable to find version in %s." % (VERSION_FILE,))
if __name__ == "__main__":
setup(
name=NAME,
version=get_version(),
description=DESCRIPTION,
license=LICENSE,
author=AUTHOR,
author_email=AUTHOR_EMAIL,
url=URL,
classifiers=CLASSIFIERS,
packages=find_packages(),
install_requires=get_requirements(REQUIREMENTS_FILE),
tests_require=get_requirements(REQUIREMENTS_DEV_FILE),
long_description=get_long_description(),
long_description_content_type="text/markdown",
include_package_data=True,
)