From 38d47e62f20fb7aef0583f2e27ffb30b3a2e5e1b Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Mon, 6 Jan 2025 18:54:50 +0530
Subject: [PATCH] fix aliasing

---
 libs/template/template.go      | 3 ++-
 libs/template/template_test.go | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/libs/template/template.go b/libs/template/template.go
index 120ca63d7..e40208445 100644
--- a/libs/template/template.go
+++ b/libs/template/template.go
@@ -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
 		}
 	}
diff --git a/libs/template/template_test.go b/libs/template/template_test.go
index 88e753816..8540b7c22 100644
--- a/libs/template/template_test.go
+++ b/libs/template/template_test.go
@@ -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)
 }