forked from metadriverse/metadrive
-
Notifications
You must be signed in to change notification settings - Fork 5
/
setup.py
130 lines (98 loc) · 3.17 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
# Please don't change the order of following packages!
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 get_version():
context = {}
with open('./metadrive/version.py', 'r') as file:
exec(file.read(), context)
return context['VERSION']
VERSION = get_version()
def is_mac():
return sys.platform == "darwin"
def is_win():
return sys.platform == "win32"
assert (3,6) <= sys.version_info < (4,), "python version >= 3.6, <4 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)
install_requires = [
"requests",
"gymnasium>=0.28",
"numpy>=1.21.6",
"matplotlib",
"pandas",
"pygame",
"tqdm",
"yapf",
"seaborn",
"tqdm",
"progressbar",
"panda3d==1.10.14",
"panda3d-gltf==0.13", # 0.14 will bring some problems
"pillow",
"pytest",
"opencv-python-headless",
"lxml",
"scipy",
"psutil",
"geopandas",
"shapely",
"filelock",
"Pygments",
]
cuda_requirement = [
"cuda-python==12.0.0",
"PyOpenGL==3.1.6",
"PyOpenGL-accelerate==3.1.6",
"pyrr==0.10.3",
"glfw",
]
gym_requirement = [
"gym>=0.19.0, <=0.26.0"
]
ros_requirement = [
"zmq"
]
setup(
name="metadrive-simulator",
python_requires='>=3.6, <4', # do version check with assert
version=VERSION,
description="An open-ended driving simulator with infinite scenes",
url="https://github.com/metadriverse/metadrive",
author="MetaDrive Team",
author_email="[email protected], [email protected]",
packages=packages,
install_requires=install_requires,
extras_require={
"cuda": cuda_requirement,
"gym": gym_requirement,
"ros": ros_requirement,
},
include_package_data=True,
license="Apache 2.0",
long_description=long_description,
long_description_content_type='text/markdown',
)
"""
How to publish to pypi and Draft github Release? Noted by Zhenghao and Quanyi in Dec 27, 2020.
0. Checkout a new branch from main called releases/x.y.z
1. Rename VERSION in metadrive/version.py to x.y.z
2. Revise the version in metadrive/assets/version.txt to x.y.z, and compress the folder: zip -r assets.zip assets
3. Commit changes and push this branch to remote
4. Remove old files and ext_modules from setup() to get a clean wheel for all platforms in py3-none-any.wheel
rm -rf dist/ build/ documentation/build/ metadrive_simulator.egg-info/ docs/build/
5. Get wheel
python setup.py sdist bdist_wheel
6. Upload to production channel
twine upload dist/*
7. Draft a release on github with new version number and upload assets.zip and the generated .whl files to the release.
8. Publish the release
9. Merge this branch into main
!!!!!!!!!!!!! NOTE: please make sure that unzip assets.zip will generate a folder called assets instead of files
"""