move model retrain to runners closes #94

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
 		}

logic/tasks/utils/utils.go

@@ -38,6 +38,7 @@ type TaskType int
 const (
 	TASK_TYPE_CLASSIFICATION TaskType = 1 + iota
 	TASK_TYPE_TRAINING
+	TASK_TYPE_RETRAINING
 )
 
 func (t Task) UpdateStatus(base BasePack, status TaskStatus, message string) (err error) {