test update

test/test_modules/test_training.py

@@ -156,7 +156,7 @@ class TestTraining:
         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")
-        obj.data_store.set("epochs", 2, "general.model")
+        obj.data_store.set("epochs", 2, "general")
         clbk, hist, lr = callbacks
         obj.data_store.set("callbacks", clbk, "general.model")
         obj.data_store.set("lr_decay", lr, "general.model")