From 9d24150031bf021ee2cbb91ae388d62b681b1969 Mon Sep 17 00:00:00 2001
From: cneud <952378+cneud@users.noreply.github.com>
Date: Tue, 19 Mar 2024 22:20:19 +0100
Subject: [PATCH] use tf1.compat mode for keras

---
 eynollah/eynollah/train/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/eynollah/eynollah/train/train.py b/eynollah/eynollah/train/train.py
index 0881182..ec6900c 100644
--- a/eynollah/eynollah/train/train.py
+++ b/eynollah/eynollah/train/train.py
@@ -1,7 +1,7 @@
 import os
 import sys
 import tensorflow as tf
-from keras.backend.tensorflow_backend import set_session
+from tensorflow.compat.v1.keras.backend import set_session
 import keras, warnings
 from keras.optimizers import *
 from sacred import Experiment