move model retrain to runners closes #94

2024-04-16 17:48:52 +01:00
parent f182b205f8
commit 06642dcb1e
9 changed files with 165 additions and 96 deletions
--- a/logic/tasks/runner/runner.go
+++ b/logic/tasks/runner/runner.go
@@ -63,6 +63,14 @@ func runner(config Config, db *sql.DB, task_channel chan Task, index int, back_c
 				logger.Error("Failed to tain the model", "error", err)
 			}

+			back_channel <- index
+			continue
+		} else if task.TaskType == int(TASK_TYPE_RETRAINING) {
+			logger.Info("Retraining Task")
+			if err = RunTaskRetrain(base, task); err != nil {
+				logger.Error("Failed to tain the model", "error", err)
+			}
+
 			back_channel <- index
 			continue
 		}