wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 18:44:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Default to TensorboardX and include in requirements. (#6836)

Browse Source
This commit is contained in:
Richard Liaw
2020-01-19 01:49:33 -08:00
committed by GitHub
parent a229bdf272
commit 341ddd0a09
4 changed files with 32 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/setup.py
+9 -5
View File
@@ -72,16 +72,20 @@ if "RAY_USE_NEW_GCS" in os.environ and os.environ["RAY_USE_NEW_GCS"] == "on":
]
extras = {
"rllib": [
"pyyaml", "gym[atari]", "opencv-python-headless", "lz4", "scipy",
"tabulate"
],
"debug": ["psutil", "setproctitle", "py-spy >= 0.2.0"],
"dashboard": ["aiohttp", "google", "grpcio", "psutil", "setproctitle"],
"serve": ["uvicorn", "pygments", "werkzeug", "flask", "pandas", "blist"],
"tune": ["tabulate"],
"tune": ["tabulate", "tensorboardX"],
}
extras["rllib"] = extras["tune"] + [
"pyyaml",
"gym[atari]",
"opencv-python-headless",
"lz4",
"scipy",
]
extras["all"] = list(set(chain.from_iterable(extras.values())))