From 0bb918f2b1cb37f66180c442f308efa7467c0247 Mon Sep 17 00:00:00 2001 From: Edward Oakes Date: Wed, 22 Apr 2020 15:54:42 -0500 Subject: [PATCH] Disable eager execution to fix test_tensorflow (#8133) --- python/ray/tests/test_tensorflow.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/ray/tests/test_tensorflow.py b/python/ray/tests/test_tensorflow.py index b26474c81..d243deaae 100644 --- a/python/ray/tests/test_tensorflow.py +++ b/python/ray/tests/test_tensorflow.py @@ -1,8 +1,10 @@ from numpy.testing import assert_almost_equal -import tensorflow.compat.v1 as tf import ray import ray.experimental.tf_utils +from ray.rllib.utils.framework import try_import_tf + +tf = try_import_tf() def make_linear_network(w_name=None, b_name=None):