Commit 6605b703 authored by lukas leufen's avatar lukas leufen

adjust two tests

parent ff32ac91
Pipeline #39353 passed with stages
in 7 minutes and 39 seconds
......@@ -27,7 +27,7 @@ def execute_removing(path):
if "data" in list_dir and path != os.path.dirname(__file__): # do not delete data folder in src
shutil.rmtree(os.path.join(path, "data"), ignore_errors=True)
# remove TestExperiment folders
remove_files_from_regex(list_dir, path, re.compile(r"TestExperiment_.*"))
remove_files_from_regex(list_dir, path, re.compile(r"TestExperiment.*"))
# remove all tracking json
remove_files_from_regex(list_dir, path, re.compile(r"tracking_\d*\.json"))
# remove all tracking pdf
......
......@@ -88,7 +88,7 @@ class TestModelSetup:
setup_with_gen.build_model()
assert isinstance(setup_with_gen.model, AbstractModelClass)
expected = {"window_history_size", "window_lead_time", "channels", "dropout_rate", "regularizer", "initial_lr",
"optimizer", "epochs", "batch_size", "activation"}
"optimizer", "epochs", "activation"}
assert expected <= self.current_scope_as_set(setup_with_gen)
def test_set_channels(self, setup_with_gen_tiny):
......
......@@ -155,7 +155,7 @@ class TestTraining:
obj.data_store.set("model", model, "general.model")
obj.data_store.set("model_path", model_path, "general")
obj.data_store.set("model_name", os.path.join(model_path, "test_model.h5"), "general.model")
obj.data_store.set("batch_size", 256, "general.model")
obj.data_store.set("batch_size", 256, "general")
obj.data_store.set("epochs", 2, "general.model")
clbk, hist, lr = callbacks
obj.data_store.set("callbacks", clbk, "general.model")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment