From a06cd681ef96c9a5f9756ec543a85ede553487c4 Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Thu, 19 Sep 2019 17:33:29 +0200 Subject: [PATCH] use tf.compat.v1 session --- tests/test_tfmad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tfmad.py b/tests/test_tfmad.py index a7f3cca..8d14fa0 100644 --- a/tests/test_tfmad.py +++ b/tests/test_tfmad.py @@ -275,7 +275,7 @@ def test_tfmad_gradient_check_tensorflow_native(with_offsets, with_cuda, gradien out_tensor = auto_diff.create_tensorflow_op(use_cuda=with_cuda, backend='tensorflow_native')(a=a_tensor, b=b_tensor) - with tf.Session() as sess: + with tf.compat.v1.Session() as sess: sess.run(tf.global_variables_initializer()) sess.run(out_tensor) -- GitLab