From 4574ee9fe0e660e6035c2c4f07251938ce668e2b Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Thu, 12 May 2022 17:20:18 +0200
Subject: [PATCH] update missing parameter in test
---
test/test_run_modules/test_training.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/test/test_run_modules/test_training.py b/test/test_run_modules/test_training.py
index 1b83b382..29717674 100644
--- a/test/test_run_modules/test_training.py
+++ b/test/test_run_modules/test_training.py
@@ -360,7 +360,7 @@ class TestTraining:
assert len(glob.glob(os.path.join(path, "plots", "TestExperiment_history_*.pdf"))) == 0
history.model.output_names = mock.MagicMock(return_value=["Main"])
history.model.metrics_names = mock.MagicMock(return_value=["loss", "mean_squared_error"])
- init_without_run.create_monitoring_plots(history, learning_rate)
+ init_without_run.create_monitoring_plots(history, learning_rate, epoch_best=1)
assert len(glob.glob(os.path.join(path, "plots", "TestExperiment_history_*.pdf"))) == 2
def test_resume_training1(self, path: str, model_path, batch_path, data_collection, statistics_per_var,
--
GitLab