diff --git a/task.go b/task.go index 5308f42..139fa31 100644 --- a/task.go +++ b/task.go @@ -409,7 +409,7 @@ func (t *AuroraTask) BuildThermosPayload() error { t.thermos.disk(*t.resources[DISK].DiskMb) } - if t.resources[GPU]!= nil && t.resources[GPU].NumGpus != nil { + if t.resources[GPU] != nil && t.resources[GPU].NumGpus != nil { t.thermos.gpu(*t.resources[GPU].NumGpus) }