From 62e9fbb1aafbf222c3f072b913bb0abb75367029 Mon Sep 17 00:00:00 2001
From: Lauren Yu <6631887+laurenyu@users.noreply.github.com>
Date: Tue, 7 May 2019 09:57:09 -0700
Subject: [PATCH] fix: adjust Ray test script for Ray 0.6.5

---
 tests/data/ray_cartpole/train_ray.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/data/ray_cartpole/train_ray.py b/tests/data/ray_cartpole/train_ray.py
index 3fe2040010..e5ccf9df95 100644
--- a/tests/data/ray_cartpole/train_ray.py
+++ b/tests/data/ray_cartpole/train_ray.py
@@ -5,7 +5,7 @@
 from ray.tune.logger import pretty_print
 
 # Based on https://github.com/ray-project/ray/blob/master/doc/source/rllib-training.rst#python-api
-ray.init(redirect_output=False, redirect_worker_output=False)
+ray.init(log_to_driver=False)
 config = ppo.DEFAULT_CONFIG.copy()
 config["num_gpus"] = int(os.environ.get("SM_NUM_GPUS", 0))
 checkpoint_dir = os.environ.get("SM_MODEL_DIR", '/Users/nadzeya/gym')