Addressing feedback comments. Running code through goimports.

This commit is contained in:
Renan DelValle 2019-01-08 11:06:08 -08:00
parent 25f5ff9f07
commit f39551af14
No known key found for this signature in database
GPG key ID: C240AD6D6F443EC9
2 changed files with 8 additions and 16 deletions

13
job.go
View file

@ -96,15 +96,8 @@ func NewJob() Job {
ramMb := aurora.NewResource()
diskMb := aurora.NewResource()
resources := make(map[ResourceType]*aurora.Resource)
resources[CPU] = numCpus
resources[RAM] = ramMb
resources[DISK] = diskMb
taskConfig.Resources = make(map[*aurora.Resource]bool)
taskConfig.Resources[numCpus] = true
taskConfig.Resources[ramMb] = true
taskConfig.Resources[diskMb] = true
resources := map[ResourceType]*aurora.Resource{CPU: numCpus, RAM: ramMb, DISK: diskMb}
taskConfig.Resources = map[*aurora.Resource]bool{numCpus: true, ramMb: true, diskMb: true}
numCpus.NumCpus = new(float64)
ramMb.RamMb = new(int64)
@ -178,7 +171,7 @@ func (j *AuroraJob) Disk(disk int64) Job {
}
func (j *AuroraJob) GPU(gpus int64) Job {
if _,ok := j.resources[GPU]; !ok {
if _, ok := j.resources[GPU]; !ok {
numGPUs := &aurora.Resource{NumGpus: new(int64)}
j.resources[GPU] = numGPUs
j.TaskConfig().Resources[numGPUs] = true