Skip to content

Commit ee3f246

Browse files
authored
fix: adjust Ray test script for Ray 0.6.5 (#781)
1 parent acbe7cc commit ee3f246

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/data/ray_cartpole/train_ray.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from ray.tune.logger import pretty_print
66

77
# Based on https://github.com/ray-project/ray/blob/master/doc/source/rllib-training.rst#python-api
8-
ray.init(redirect_output=False, redirect_worker_output=False)
8+
ray.init(log_to_driver=False)
99
config = ppo.DEFAULT_CONFIG.copy()
1010
config["num_gpus"] = int(os.environ.get("SM_NUM_GPUS", 0))
1111
checkpoint_dir = os.environ.get("SM_MODEL_DIR", '/Users/nadzeya/gym')

0 commit comments

Comments
 (0)