mod notebook interface

models/cloudbrain.go

@@ -301,6 +301,13 @@ type CreateNotebookParams struct {
 	Flavor      string     		`json:"flavor"`
 	Spec  		Spec 			`json:"spec"`
 	Workspace   Workspace   	`json:"workspace"`
+	Pool		Pool			`json:"pool"`
+}
+
+type Pool struct {
+	ID    	string     		`json:"id"`
+	Name 	string       	`json:"name"`
+	Type	string     		`json:"type"`
 }
 
 type Workspace struct {

modules/modelarts/modelarts.go

@@ -13,7 +13,11 @@ const (
 	storageTypeOBS = "obs"
 	autoStopDuration = 4 * 60 * 60
 	flavor = "modelarts.kat1.xlarge"
-	profileID = "Python3-ascend910-arm"
+	//profileID = "Python3-ascend910-arm"
+	profileID = "efa847c0-7359-11eb-b34f-0255ac100057"
+	poolID = "pool1328035d"
+	poolName = "train-private-1"
+	poolType = "USER_DEFINED"
 
 	DataSetMountPath = "/home/ma-user/work"
 	NotebookEnv = "Python3"
@@ -28,6 +32,11 @@ func GenerateTask(ctx *context.Context, jobName, uuid, description string) error
 		Description:description,
 		ProfileID:  profileID,
 		Flavor: flavor,
+		Pool: models.Pool{
+			ID: poolID,
+			Name: poolName,
+			Type: poolType,
+		},
 		Spec: models.Spec{
 			Storage: models.Storage{
 				Type: storageTypeOBS,