forked from metadriverse/scenarionet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
97 lines (82 loc) · 2.38 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import os
import sys
from os import path
from setuptools import setup, find_namespace_packages # This should be place at top!
ROOT_DIR = os.path.dirname(__file__)
def is_mac():
return sys.platform == "darwin"
def is_win():
return sys.platform == "win32"
assert sys.version_info.major == 3 and 6 <= sys.version_info.minor < 12, \
"python version >= 3.6, <3.12 is required"
this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
packages = find_namespace_packages(
exclude=("docs", "docs.*", "documentation", "documentation.*", "build.*"))
print("We will install the following packages: ", packages)
""" ===== Remember to modify the EDITION at first ====="""
version = "0.0.1"
install_requires = [
"numpy>=1.23.0",
"matplotlib",
"pandas",
"tqdm",
"metadrive-simulator>=0.4.1.2",
"geopandas",
"yapf",
"shapely"
]
doc = [
"sphinxemoji",
"sphinx",
"sphinx_rtd_theme",
]
train_requirement = [
"ray[rllib]==1.0.0",
# "torch",
"wandb==0.12.1",
"aiohttp==3.6.0",
"gymnasium",
"tensorflow",
"tensorflow_probability"
]
# Remove the dependencies to real-world dataset. Instead, we will point the user to the installation guideline
# in the original sources.
#
# waymo = ["waymo-open-dataset-tf-2-11-0", "tensorflow==2.11.0"]
#
# nuplan = ["nuplan-devkit>=1.2.0",
# "bokeh==2.4",
# "hydra-core",
# "chardet",
# "pyarrow",
# "aiofiles",
# "retry",
# "boto3",
# "aioboto3"]
#
# nuscenes = ["nuscenes-devkit>=1.1.10"]
setup(
name="scenarionet",
python_requires='>=3.8', # do version check with assert
version=version,
description="Scalable Traffic Scenario Management System",
url="https://github.com/metadriverse/ScenarioNet",
author="MetaDrive Team",
author_email="[email protected], [email protected]",
packages=packages,
install_requires=install_requires,
extras_require={
"train": train_requirement,
"doc": doc,
# "waymo": waymo,
# "nuplan": nuplan,
# "nuscenes": nuscenes,
# "all": nuscenes + waymo + nuplan
},
include_package_data=True,
license="Apache 2.0",
long_description=long_description,
long_description_content_type='text/markdown',
)