steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix aliasing

This commit is contained in:
Shreyas Goenka 2025-01-06 18:54:50 +05:30
parent 57a75190cc
commit 38d47e62f2
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/template/template.go
View File

@ -3,6 +3,7 @@ package template
import (
"context"
"fmt"
"slices"
"strings"
"github.com/databricks/cli/libs/cmdio"
@ -120,7 +121,7 @@ func SelectTemplate(ctx context.Context) (TemplateName, error) {
func Get(name TemplateName) *Template {
for _, template := range allTemplates {
if template.name == name {
if template.name == name || slices.Contains(template.aliases, string(name)) {
return &template
}
}

3
libs/template/template_test.go
View File

@ -86,4 +86,7 @@ func TestTemplateGet(t *testing.T) {
tmpl := Get(TemplateName(name))
assert.Nil(t, tmpl)
}
// Assert the alias works.
assert.Equal(t, Get(TemplateName("mlops-stack")).name, MlopsStacks)
}