-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
38 lines (34 loc) · 946 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
33
34
35
36
37
38
#!/usr/bin/env python
from intervention import __version__
try:
from setuptools import setup
except ImportError:
from distutils.core import setup
setup(
name="intervention",
version=__version__,
description="A simulator implementation of learning from interventions during autonomous driving",
author="Thomas Churchman",
author_email="thomas@kepow.org",
packages=["intervention",],
install_requires=[
# Generic:
"typing-extensions~=3.7",
"loguru~=0.4",
"click~=7.0",
"dataclass-csv~=1.1",
# Image processing / models:
"tensorboard~=2.4",
"torch~=1.5",
"torchvision~=0.7",
"opencv-python~=4.0",
"pygame >=1.9, <3.0a0",
"pillow~=8.1",
"matplotlib~=3.0",
# Computation:
"numpy~=1.20",
"networkx~=2.5",
"pandas~=1.2",
],
scripts=["scripts/intervention-learning"],
)