From be55310cc9640875c7e30ecc114193c439581f0d Mon Sep 17 00:00:00 2001
From: shreyas-goenka <88374338+shreyas-goenka@users.noreply.github.com>
Date: Wed, 13 Sep 2023 19:57:31 +0200
Subject: [PATCH] Use enums for default python template (#765)

## Changes
This PR changes schema to use the enum type for the default template
yes/no questions.

## Tests
Manually
---
 .../default-python/databricks_template_schema.json       | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/libs/template/templates/default-python/databricks_template_schema.json b/libs/template/templates/default-python/databricks_template_schema.json
index 22c65f30..db8adcce 100644
--- a/libs/template/templates/default-python/databricks_template_schema.json
+++ b/libs/template/templates/default-python/databricks_template_schema.json
@@ -7,26 +7,23 @@
             "order": 1
         },
         "include_notebook": {
-            "todo": "use an enum here, see https://github.com/databricks/cli/pull/668",
             "type": "string",
             "default": "yes",
-            "pattern": "^(yes|no)$",
+            "enum": ["yes", "no"],
             "description": "Include a stub (sample) notebook in 'my_project/src'",
             "order": 2
         },
         "include_dlt": {
-            "todo": "use an enum here, see https://github.com/databricks/cli/pull/668",
             "type": "string",
             "default": "yes",
-            "pattern": "^(yes|no)$",
+            "enum": ["yes", "no"],
             "description": "Include a stub (sample) DLT pipeline in 'my_project/src'",
             "order": 3
         },
         "include_python": {
-            "todo": "use an enum here, see https://github.com/databricks/cli/pull/668",
             "type": "string",
             "default": "yes",
-            "pattern": "^(yes|no)$",
+            "enum": ["yes", "no"],
             "description": "Include a stub (sample) Python package 'my_project/src'",
             "order": 4
         }