diff --git a/routers/api/v1/repo/modelarts.go b/routers/api/v1/repo/modelarts.go index 4bdd73a19..93ea69431 100755 --- a/routers/api/v1/repo/modelarts.go +++ b/routers/api/v1/repo/modelarts.go @@ -241,9 +241,14 @@ func TrainJobGetLog(ctx *context.APIContext) { return } - task, err := models.GetCloudbrainByID(id) + temp, err := models.GetCloudbrainByID(id) + if err != nil { + log.Error("GetCloudbrainByID(%s) failed:%v", id, err.Error()) + return + } + task, err := models.GetCloudbrainByJobIDAndVersionName(temp.JobID, versionName) if err != nil { - log.Error("GetCloudbrainByJobID(%s) failed:%v", id, err.Error()) + log.Error("GetCloudbrainByID(%s) failed:%v", id, err.Error()) return } resultLogFile, result, err := trainJobGetLogContent(task.JobID, task.VersionID, baseLine, order, lines_int)