From 43e54f8dba5fcc2b4957e18a9e82c324c64f03d5 Mon Sep 17 00:00:00 2001
From: Roman Ivaniuk <rivan@equinor.com>
Date: Wed, 26 Jun 2024 16:58:47 +0300
Subject: [PATCH] Fix: pin `keras` to `3.3.3` due to a bug in `keras==3.4.0`
 (#1386)

---
 requirements/full_requirements.txt | 4 ++--
 requirements/requirements.in       | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/requirements/full_requirements.txt b/requirements/full_requirements.txt
index d66e6a5b5..db923af68 100644
--- a/requirements/full_requirements.txt
+++ b/requirements/full_requirements.txt
@@ -184,8 +184,9 @@ joblib==1.3.2
     # via scikit-learn
 jsonpickle==3.0.2
     # via azureml-core
-keras==3.4.0
+keras==3.3.3
     # via
+    #   -r requirements.in
     #   scikeras
     #   tensorflow
 kiwisolver==1.4.5
@@ -278,7 +279,6 @@ packaging==21.3
     #   -r requirements.in
     #   azureml-core
     #   docker
-    #   keras
     #   knack
     #   marshmallow
     #   matplotlib
diff --git a/requirements/requirements.in b/requirements/requirements.in
index a4d27b9b5..d8a75fe08 100644
--- a/requirements/requirements.in
+++ b/requirements/requirements.in
@@ -5,6 +5,8 @@ jinja2~=3.1
 python-dateutil~=2.8
 tensorflow~=2.16.0
 scikeras~=0.13.0
+# There's a bug in keras 3.4.0 with loading models (https://github.com/keras-team/keras/issues/19921)
+keras<3.4.0
 Flask>=2.2.5,<3.0.0
 simplejson~=3.17
 catboost~=1.2.2