Disable eager execution to fix test_tensorflow (#8133)

2026-06-30 17:58:35 +08:00 · 2020-04-22 15:54:42 -05:00
parent f9f41e5a1a
commit 0bb918f2b1
1 changed files with 3 additions and 1 deletions
@@ -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):