diff --git a/modules/timer/timer.go b/modules/timer/timer.go index a0220380c..03bda00d2 100755 --- a/modules/timer/timer.go +++ b/modules/timer/timer.go @@ -18,6 +18,9 @@ func LaunchCronJob() { specCheckBlockChainUserSuccess := "*/10 * * * *" c.AddFunc(specCheckBlockChainUserSuccess, repo.HandleBlockChainUnSuccessUsers) + specModelSafetySuccess := "*/10 * * * *" + c.AddFunc(specModelSafetySuccess, repo.TimerHandleModelSafetyTestTask) + specCheckRepoBlockChainSuccess := "*/1 * * * *" c.AddFunc(specCheckRepoBlockChainSuccess, repo.HandleBlockChainUnSuccessRepos) diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 4569b3aae..50379fe08 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -2118,7 +2118,7 @@ func CloudBrainBenchmarkIndex(ctx *context.Context) { } var jobTypes []string - jobTypes = append(jobTypes, string(models.JobTypeBenchmark), string(models.JobTypeBrainScore), string(models.JobTypeSnn4imagenet)) + jobTypes = append(jobTypes, string(models.JobTypeBenchmark), string(models.JobTypeBrainScore), string(models.JobTypeSnn4imagenet), string(models.JobTypeModelSafety)) ciTasks, count, err := models.Cloudbrains(&models.CloudbrainsOptions{ ListOptions: models.ListOptions{ Page: page,