Skip to content

Commit

Permalink
Typo in extension
Browse files Browse the repository at this point in the history
  • Loading branch information
dgruber committed Mar 29, 2023
1 parent 635cfac commit 600ab00
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions jobtemplate_extensions.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ const (
ExtensionProlog = "prolog"
ExtensionEpilog = "epilog"
ExtensionSpot = "spot"
ExctensionAccelerators = "accelerators"
ExtensionAccelerators = "accelerators"
ExtensionTasksPerNode = "tasks_per_node"
ExtensionDockerOptions = "docker_options"
)
Expand Down Expand Up @@ -78,7 +78,7 @@ func GetAcceleratorsExtension(jt drmaa2interface.JobTemplate) (string, int64, bo
if jt.ExtensionList == nil {
return "", 0, false
}
extension, hasExtensions := jt.ExtensionList[ExctensionAccelerators]
extension, hasExtensions := jt.ExtensionList[ExtensionAccelerators]
if hasExtensions {
a := strings.Split(extension, "*")
if len(a) >= 2 {
Expand All @@ -95,7 +95,7 @@ func SetAcceleratorsExtension(jt drmaa2interface.JobTemplate, count int64, accel
if jt.ExtensionList == nil {
jt.ExtensionList = make(map[string]string)
}
jt.ExtensionList[ExctensionAccelerators] = strconv.FormatInt(count, 10) + "*" + accelerator
jt.ExtensionList[ExtensionAccelerators] = strconv.FormatInt(count, 10) + "*" + accelerator
}
return jt
}
Expand Down
4 changes: 2 additions & 2 deletions jobtemplate_extensions_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -41,8 +41,8 @@ var _ = Describe("JobtemplateExtensions", func() {
It("should set the accelerators request", func() {
jt := drmaa2interface.JobTemplate{}
jt = SetAcceleratorsExtension(jt, 1, "nvidia-tesla-k80")
Expect(jt.ExtensionList).To(HaveKey(ExctensionAccelerators))
Expect(jt.ExtensionList[ExctensionAccelerators]).To(Equal("1*nvidia-tesla-k80"))
Expect(jt.ExtensionList).To(HaveKey(ExtensionAccelerators))
Expect(jt.ExtensionList[ExtensionAccelerators]).To(Equal("1*nvidia-tesla-k80"))
accelerators, count, exists := GetAcceleratorsExtension(jt)
Expect(exists).To(BeTrue())
Expect(accelerators).To(Equal("nvidia-tesla-k80"))
Expand Down

0 comments on commit 600ab00

Please sign in to comment.