-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.py
63 lines (57 loc) · 1.65 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
import sys
from setuptools import setup
try:
from setuptools_rust import RustExtension, Binding, Strip
except ImportError:
import subprocess
errno = subprocess.call([sys.executable, '-m', 'pip', 'install', 'setuptools-rust'])
if errno:
print("Please install the setuptools-rust package.")
raise SystemExit(errno)
else:
from setuptools_rust import RustExtension, Binding, Strip
try:
import toml
except ImportError:
import subprocess
errno = subprocess.call([sys.executable, '-m', 'pip', 'install', 'toml'])
if errno:
print("Please install the toml package.")
raise SystemExit(errno)
else:
import toml
cfg = toml.load("Cargo.toml")
package_meta = cfg['package']
setup_requires = [
'setuptools',
'setuptools-rust>=0.10.1'
]
install_requires = ['numpy']
setup(
name=package_meta['name'],
version=package_meta['version'],
author=package_meta['authors'][0],
rust_extensions=[
RustExtension(
'pointprocesses.temporal',
'Cargo.toml',
binding=Binding.PyO3,
strip=Strip.Debug),
RustExtension(
'pointprocesses.spatial',
'Cargo.toml',
binding=Binding.PyO3,
strip=Strip.Debug),
RustExtension(
'pointprocesses.likelihood',
'Cargo.toml',
binding=Binding.PyO3,
strip=Strip.Debug),
],
packages=['pointprocesses'],
install_requires=install_requires,
setup_requires=setup_requires,
include_package_data=True,
# rust extensions are not zip safe, just like C-extensions.
zip_safe=False
)