routers/repo/aisafety.go

@@ -186,6 +186,9 @@ func DelAiSafetyTask(ctx *context.Context) {
 	}
 	if task.Type == models.TypeCloudBrainOne {
 		cloudbrainTask.DeleteCloudbrainJobStorage(task.JobName, models.TypeCloudBrainOne)
+	} else {
+		modelarts.DelTrainJobVersion(task.JobID, strconv.FormatInt(task.VersionID, 10))
+		cloudbrainTask.DeleteCloudbrainJobStorage(task.JobName, models.TypeCloudBrainTwo)
 	}
 	err = models.DeleteJob(task)
 	if err != nil {
@@ -743,7 +746,7 @@ func createForNPU(ctx *context.Context, jobName string) error {
 		Value: "s3:/" + ckptUrl,
 	})
 	uuid := srcDataset + ";" + combatDataset
-	datasUrlList, dataUrl, datasetNames, isMultiDataset, err := getDatasUrlListByUUIDS(uuid)
+	datasUrlList, dataUrl, datasetNames, _, err := getDatasUrlListByUUIDS(uuid)
 	if err != nil {
 		return err
 	}
@@ -753,12 +756,11 @@ func createForNPU(ctx *context.Context, jobName string) error {
 		log.Error("Failed to Marshal: %v", err)
 		return err
 	}
-	if isMultiDataset {
-		param = append(param, models.Parameter{
-			Label: modelarts.MultiDataUrl,
-			Value: string(jsondatas),
-		})
-	}
+
+	param = append(param, models.Parameter{
+		Label: modelarts.MultiDataUrl,
+		Value: string(jsondatas),
+	})
 
 	existDeviceTarget := false
 	if len(Params) != 0 {