db50bca7fd
* scenarionet training * wandb * train utils * fix callback * run PPO * use pg test * save path * use torch * add dependency * update ignore * update training * large model * use curriculum training * add time to exp name * storage_path * restore * update training * use my key * add log message * check seed * restore callback * restore call bacl * add log message * add logging message * restore ray1.4 * length 500 * ray 100 * wandb * use tf * more levels * add callback * 10 worker * show level * no env horizon * callback result level * more call back * add diffuculty * add mroen stat * mroe stat * show levels * add callback * new * ep len 600 * fix setup * fix stepup * fix to 3.8 * update setup * parallel worker! * new exp * add callback * lateral dist * pg dataset * evaluate * modify config * align config * train single RL * update training script * 100w eval * less eval to reveal * 2000 env eval * new trianing * eval 1000 * update eval * more workers * more worker * 20 worker * dataset to database * split tool! * split dataset * try fix * train 003 * fix mapping * fix test * add waymo tqdm * utils * fix bug * fix bug * waymo * int type * 8 worker read * disable * read file * add log message * check existence * dist 0 * int * check num * suprass warning * add filter API * filter * store map false * new * ablation * filter * fix * update filyter * reanme to from * random select * add overlapping checj * fix * new training sceheme * new reward * add waymo train script * waymo different config * copy raw data * fix bug * add tqdm * update readme * waymo * pg * max lateral dist 3 * pg * crash_done instead of penalty * no crash done * gpu * update eval script * steering range penalty * evaluate * finish pg * update setup * fix bug * test * fix * add on line * train nuplan * generate sensor * udpate training * static obj * multi worker eval * filx bug * use ray for testing * eval! * filter senario * id filter * fox bug * dist = 2 * filter * eval * eval ret * ok * update training pg * test before use * store data=False * collect figures * capture pic --------- Co-authored-by: Quanyi Li <quanyi@bolei-gpu02.cs.ucla.edu>
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
# 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 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.21.6, <=1.24.2",
|
|
"matplotlib",
|
|
"pandas",
|
|
"tqdm",
|
|
"metadrive-simulator",
|
|
"geopandas",
|
|
"yapf==0.30.0",
|
|
"shapely"
|
|
]
|
|
|
|
train_requirement = [
|
|
"ray[rllib]==1.0.0",
|
|
# "torch",
|
|
"wandb==0.12.1",
|
|
"aiohttp==3.6.0",
|
|
"gymnasium",
|
|
"tensorflow",
|
|
"tensorflow_probability"]
|
|
|
|
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="quanyili0057@gmail.com, pzh@cs.ucla.edu",
|
|
packages=packages,
|
|
install_requires=install_requires,
|
|
extras_require={
|
|
"train": train_requirement,
|
|
},
|
|
include_package_data=True,
|
|
license="Apache 2.0",
|
|
long_description=long_description,
|
|
long_description_content_type='text/markdown',
|
|
)
|
|
|
|
"""
|
|
How to publish to pypi? Noted by Zhenghao in Dec 27, 2020.
|
|
|
|
0. Rename version in setup.py
|
|
|
|
1. 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/ scenarionet.egg-info/ docs/build/
|
|
|
|
2. Rename current version to X.Y.Z.rcA, where A is arbitrary value represent "release candidate A".
|
|
This is really important since pypi do not support renaming and re-uploading.
|
|
Rename version in setup.py
|
|
|
|
3. Get wheel
|
|
python setup.py sdist bdist_wheel
|
|
|
|
4. Upload to test channel
|
|
twine upload --repository testpypi dist/*
|
|
|
|
5. Test as next line. If failed, change the version name and repeat 1, 2, 3, 4, 5.
|
|
pip install --index-url https://test.pypi.org/simple/ scenarionet
|
|
|
|
6. Rename current version to X.Y.Z in setup.py, rerun 1, 3 steps.
|
|
|
|
7. Upload to production channel
|
|
twine upload dist/*
|
|
|
|
"""
|