modified shuguangAi implementations

Former-commit-id: ec3fe090a5

api/internal/scheduler/schedulers/option/aiOption.go

@@ -1,9 +1,11 @@
 package option
 
 type AiOption struct {
-	AiType       string // shuguangAi/octopus
+	//AiType       string // shuguangAi/octopus
 	ResourceType string // cpu/gpu/compute card
 	TaskType     string // pytorch/tensorflow
+	DatasetsType string
+	CodeType     string
 
 	ImageId    string
 	SpecId     string

api/internal/scheduler/strategy/replication.go

@@ -1,7 +1,7 @@
 package strategy
 
 import (
-	"github.com/pkg/errors"
+	"errors"
 	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/entity"
 	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/strategy/param"
 )

api/internal/storeLink/shuguangai.go

@@ -210,9 +210,14 @@ func (o *ShuguangAi) GetResourceSpecs() (*collector.ResourceSpecs, error) {
 }
 
 func (o *ShuguangAi) Execute(option *option.AiOption) (interface{}, error) {
-	task, err := o.SubmitTask(option.ImageId, option.Cmd, option.Envs, option.Params, option.ResourceId, option.AiType)
+	o.generateSubmitParams(option)
+	task, err := o.SubmitTask(option.ImageId, option.Cmd, option.Envs, option.Params, option.ResourceId, option.TaskType)
 	if err != nil {
 		return nil, err
 	}
 	return task, nil
 }
+
+func (o *ShuguangAi) generateSubmitParams(option *option.AiOption) {
+
+}