From 2fb17877677f013e67c54bccbe35e5ed0afa0e3b Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Wed, 17 Nov 2021 14:36:38 +0100
Subject: [PATCH] update ready to run test
---
test/test_run_modules/test_training.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/test/test_run_modules/test_training.py b/test/test_run_modules/test_training.py
index 51ea1cd3..b16c0c25 100644
--- a/test/test_run_modules/test_training.py
+++ b/test/test_run_modules/test_training.py
@@ -194,6 +194,7 @@ class TestTraining:
obj.data_store.set("data_collection", data_collection, "general.val")
obj.data_store.set("data_collection", data_collection, "general.test")
obj.model.compile(**obj.model.compile_options)
+ keras.utils.get_custom_objects().update(obj.model.custom_objects)
return obj
@pytest.fixture
--
GitLab