Skip to content
Snippets Groups Projects
Commit 2fb17877 authored by leufen1's avatar leufen1
Browse files

update ready to run test

parent 5cb2ba3a
No related branches found
No related tags found
5 merge requests!413update release branch,!412Resolve "release v2.0.0",!361name of pdf starts now with feature_importance, there is now also another...,!350Resolve "upgrade code to TensorFlow V2",!335Resolve "upgrade code to TensorFlow V2"
Pipeline #83529 passed
Stage: test
Stage: docs
Stage: pages
Stage: deploy
Show whitespace changes
Inline Side-by-side
test/test_run_modules/test_training.py
+ 1
− 0
View file @ 2fb17877
...@@ -194,6 +194,7 @@ class TestTraining: ...@@ -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.val")
obj.data_store.set("data_collection", data_collection, "general.test") obj.data_store.set("data_collection", data_collection, "general.test")
obj.model.compile(**obj.model.compile_options) obj.model.compile(**obj.model.compile_options)
keras.utils.get_custom_objects().update(obj.model.custom_objects)
return obj return obj
@pytest.fixture @pytest.fixture
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment