diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go index 99aa5d53..b5dd1407 100644 --- a/internal/storeLink/modelarts.go +++ b/internal/storeLink/modelarts.go @@ -161,6 +161,13 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri inputs := make([]*modelarts.InputTraining, 0) outputs := make([]*modelarts.OutputTraining, 0) outputValue := "" + + // add default param, env + current_id := strconv.Itoa(int(m.participantId)) + current_platform := CURRENT_PLATFORM + COMMA + current_id + params = append(params, current_platform) + envs = append(envs, current_platform) + for _, env := range envs { // 找到第一个逗号位置 idx := strings.Index(env, COMMA) diff --git a/internal/storeLink/openi.go b/internal/storeLink/openi.go index 2f179811..0403c2e8 100644 --- a/internal/storeLink/openi.go +++ b/internal/storeLink/openi.go @@ -216,6 +216,11 @@ func (o *OpenI) SubmitTask(ctx context.Context, imageId string, cmd string, envs } `json:"parameter"` } + // add default param + current_id := strconv.Itoa(int(o.participantId)) + current_platform := CURRENT_PLATFORM + COMMA + current_id + params = append(params, current_platform) + for _, param := range params { s := strings.Split(param, COMMA) st := struct { diff --git a/internal/storeLink/storeLink.go b/internal/storeLink/storeLink.go index d75c56df..6879a0bc 100644 --- a/internal/storeLink/storeLink.go +++ b/internal/storeLink/storeLink.go @@ -88,6 +88,7 @@ const ( POINT = "point" RUNNINGTASK = "RUNNING_TASK" RUNNING = "RUNNING" + CURRENT_PLATFORM = "CURRENT_PLATFORM" //default param ) var (