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