[autoscaler/tune] Optional YAML Fields + Fix Pretty Printing for Tune (#1541)

This commit is contained in:
Richard Liaw
2018-03-04 23:35:58 -08:00
committed by GitHub
parent 061e435411
commit 162d063f0d
12 changed files with 199 additions and 70 deletions
+6
View File
@@ -33,6 +33,10 @@ ray_ui_files = [
"ray/core/src/catapult_files/trace_viewer_full.html"
]
ray_autoscaler_files = [
"ray/autoscaler/aws/example-full.yaml"
]
# The UI files are mandatory if the INCLUDE_UI environment variable equals 1.
# Otherwise, they are optional.
if "INCLUDE_UI" in os.environ and os.environ["INCLUDE_UI"] == "1":
@@ -40,6 +44,8 @@ if "INCLUDE_UI" in os.environ and os.environ["INCLUDE_UI"] == "1":
else:
optional_ray_files += ray_ui_files
optional_ray_files += ray_autoscaler_files
extras = {
"rllib": [
"tensorflow", "pyyaml", "gym[atari]", "opencv-python",