From 55b214c23d337f73c88f54fe3c662ed253146e51 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 12 May 2021 15:42:59 +0300
Subject: [PATCH] Fix direct import of keras save/load model functions (issue
 #355)

---
 clearml/binding/frameworks/tensorflow_bind.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/clearml/binding/frameworks/tensorflow_bind.py b/clearml/binding/frameworks/tensorflow_bind.py
index fc9bd463..a3d15d16 100644
--- a/clearml/binding/frameworks/tensorflow_bind.py
+++ b/clearml/binding/frameworks/tensorflow_bind.py
@@ -1558,6 +1558,12 @@ class PatchKerasModelIO(object):
                     keras_hdf5.load_weights_from_hdf5_group, PatchKerasModelIO._load_weights)
                 keras_hdf5.load_weights_from_hdf5_group_by_name = _patched_call(
                     keras_hdf5.load_weights_from_hdf5_group_by_name, PatchKerasModelIO._load_weights)
+                if hasattr(keras_hdf5, 'load_model_from_hdf5'):
+                    keras_hdf5.load_model_from_hdf5 = _patched_call(
+                        keras_hdf5.load_model_from_hdf5, PatchKerasModelIO._load_model)
+                if hasattr(keras_hdf5, 'save_model_to_hdf5'):
+                    keras_hdf5.save_model_to_hdf5 = _patched_call(
+                        keras_hdf5.save_model_to_hdf5, PatchKerasModelIO._save_model)
 
         except Exception as ex:
             LoggerRoot.get_base_logger(TensorflowBinding).warning(str(ex))