-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
32 lines (30 loc) · 985 Bytes
/
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
from setuptools import setup, Extension
from Cython.Build import cythonize
extensions = [
Extension("sloop.oopomdp.models.components.grid_map",
["sloop/oopomdp/models/components/grid_map.pyx"]),
]
setup(name='sloop',
packages=['sloop'],
version='0.1',
description='Spatial Language OOPOMDP',
install_requires=[
"pomdp-py==1.2.4.5",
"numpy>=1.20.3",
"opencv-contrib-python>=4.5.1.48",
"opencv-python>=4.1.1.26",
"pandas>=1.1.2",
"Pillow>=8.2.0",
"pygame>=1.9.6",
"scipy>=1.3.3",
"seaborn>=0.11.0",
"spacy==2.2.4",
"torch==1.8.1",
"torchvision==0.9.1",
"allennlp>=2.5.0"
],
ext_modules=cythonize(extensions,
build_dir="build",
compiler_directives={"language_level": "3"}),
package_data={"sloop": ["*.pxd", "*.pyx"]},
zip_safe=False)