aiforge/modules/dataset/dataset.go

package dataset

import "code.gitea.io/gitea/models"

func GetResourceType(cloudbrainType int) string {
	if cloudbrainType == 0 {
		return "CPU/GPU"
	} else {
		return "NPU"
	}
}

func GetStatusText(isPrivate bool) string {
	if isPrivate {
		return "dataset.private"
	} else {
		return "dataset.public"
	}
}

func IsShowDataSetOfCurrentRepo(repoID int64) bool {
	repo := models.Repository{
		ID: repoID,
	}

	dataset, _ := models.GetDatasetByRepo(&repo)
	if dataset != nil {
		return true
	}
	if models.HasReferenceDataset(repoID) {
		return false
	}
	return true

}