fix at least 1 test of training module

test/test_run_modules/test_training.py

@@ -306,7 +306,7 @@ class TestTraining:
         assert init_without_run.train_set._collection.return_value == "mock_train_gen"
 
     def test_set_generators(self, init_without_run):
-        sets = ["train", "val", "test"]
+        sets = ["train", "val"]
         assert all([getattr(init_without_run, f"{obj}_set") is None for obj in sets])
         init_without_run.set_generators()
         assert not all([getattr(init_without_run, f"{obj}_set") is None for obj in sets])
@@ -366,10 +366,10 @@ class TestTraining:
     def test_resume_training1(self, path: str, model_path, batch_path, data_collection, statistics_per_var,
                               window_history_size, window_lead_time):
 
-        obj_1st = self.create_training_obj(2, path, data_collection, batch_path, model_path, statistics_per_var,
+        obj_1st = self.create_training_obj(4, path, data_collection, batch_path, model_path, statistics_per_var,
                                            window_history_size, window_lead_time)
         keras.utils.get_custom_objects().update(obj_1st.model.custom_objects)
         assert obj_1st._run() is None
-        obj_2nd = self.create_training_obj(4, path, data_collection, batch_path, model_path, statistics_per_var,
+        obj_2nd = self.create_training_obj(8, path, data_collection, batch_path, model_path, statistics_per_var,
                                            window_history_size, window_lead_time)
         assert obj_2nd._run() is None