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

3
libs/template/template_test.go
View File

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