final

my_mlops_project/databricks.yml

@@ -9,6 +9,8 @@ variables:
     description: Source catalog for the model
   target_catalog:
     description: Source catalog for the model
+  branch_name:
+    description: project branch to use in the code
 
 include:
   - resources/*.yml
@@ -23,6 +25,8 @@ targets:
     default: true
     variables:
       source_catalog: mlops_dev
+      target_catalog: mlops_stage
+      branch_name: main
     workspace:
       host: https://dbc-325dd65c-6716.cloud.databricks.com
 
@@ -35,6 +39,7 @@ targets:
     variables:
       source_catalog: mlops_dev
       target_catalog: mlops_stage
+      branch_name: main
     permissions:
       - user_name: [email protected]
         level: CAN_MANAGE
@@ -48,6 +53,7 @@ targets:
     variables:
       source_catalog: mlops_stage
       target_catalog: mlops_prod
+      branch_name: release
     permissions:
       - user_name: [email protected]
         level: CAN_MANAGE

my_mlops_project/resources/forecast_model_validation_and_migration.job.yml

@@ -13,17 +13,21 @@ resources:
       parameters:
         - name: model_name
           default: interest_forecast_7_day
+        - name: model_version
+          default: 1
         - name: source_catalog
           default: ${var.source_catalog}
         - name: source_db
           default: interest_forecast
         - name: validation_data
           default: validation_table
+        - name: target_catalog
+          default: ${var.target_catalog}
       email_notifications: {}
       run_as:
         user_name: [email protected]
       git_source:
-        git_branch: main
+        git_branch: ${var.branch_name}
         git_provider: gitHub
         git_url: https://github.com/himanshu3jul/model_validation_deployment
       max_concurrent_runs: 1