Browse Source

提交代码,解决导入新模型时选择的顺序问题

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.21.12.1^2
zouap 3 years ago
parent
commit
3d59d23070
1 changed files with 11 additions and 2 deletions
  1. +11
    -2
      models/cloudbrain.go

+ 11
- 2
models/cloudbrain.go View File

@@ -964,12 +964,21 @@ func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) {
)

cloudbrains := make([]*CloudbrainInfo, 0)
if err := sess.Select("job_id,job_name").Table(&Cloudbrain{}).Where("job_id in (?)", sess.Distinct("job_id").Table(&Cloudbrain{}).Where(cond)).OrderBy("created_unix DESC").
if err := sess.Select("job_id,job_name").Table(&Cloudbrain{}).Where(cond).OrderBy("created_unix DESC").
Find(&cloudbrains); err != nil {
return nil, 0, fmt.Errorf("Find: %v", err)
}

return cloudbrains, int(len(cloudbrains)), nil
keys := make(map[string]string)
uniqueElements := make([]*CloudbrainInfo, 0)
for _, entry := range cloudbrains {
if _, value := keys[entry.JobID]; !value {
keys[entry.JobID] = entry.JobName
uniqueElements = append(uniqueElements, entry)
}
}

return uniqueElements, int(len(uniqueElements)), nil
}

func CloudbrainsVersionList(opts *CloudbrainsOptions) ([]*CloudbrainInfo, int, error) {


Loading…
Cancel
Save