From 8b7fd2dc32bdc492257a6fa04e64d5ad812db24d Mon Sep 17 00:00:00 2001
From: Daniel Falbel <dfalbel@gmail.com>
Date: Thu, 2 Jul 2020 14:29:25 -0300
Subject: [PATCH] after 2.0 we should not set session.

---
 R/seed.R | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/R/seed.R b/R/seed.R
index cdc9322c4..456c8d6bd 100644
--- a/R/seed.R
+++ b/R/seed.R
@@ -10,7 +10,9 @@ tensorflow_on_before_use_session <- function(quiet) {
 }
 
 tensorflow_on_use_session <- function(sess, quiet) {
-  if (is_backend("tensorflow"))
-    keras$backend$set_session(sess)
+  if (is_backend("tensorflow")) {
+    if (tensorflow::tf_version() < "2.0")
+      keras$backend$set_session(sess)
+  }
 }