Use enum for selection type input parameters

databricks / mlops-stacks

This repo provides a customizable stack for starting new ML projects on Databricks that follow production best-practices out of the box.

https://docs.databricks.com/en/dev-tools/bundles/mlops-stacks.html

Apache License 2.0

425 stars 143 forks source link

Use enum for selection type input parameters #110

Closed shreyas-goenka closed 10 months ago

shreyas-goenka commented 10 months ago

Changes

This PR switches to enum fields for relevant input parameters. This allows for a better selection UX.

Note, this PR depends on a CLI version which would have this PR included: https://github.com/databricks/cli/pull/916. We will need to first set a correct min_cli_version before merging this.

Tests

Tested Manually. The video below shows what the initialization experience looks like now:

https://github.com/databricks/mlops-stacks/assets/88374338/e2670471-f953-41d2-b9c7-0c6e08c4c43c