From 435f682e88b5624b4ec72719d161e1d5e6e38480 Mon Sep 17 00:00:00 2001
From: teocalvo <teocalvo>
Date: Wed, 7 Aug 2024 14:08:37 +0000
Subject: [PATCH] Model taks

---
 src/workflows/ds-bricks.json | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/workflows/ds-bricks.json b/src/workflows/ds-bricks.json
index aca2890..e42e9b0 100644
--- a/src/workflows/ds-bricks.json
+++ b/src/workflows/ds-bricks.json
@@ -96,6 +96,34 @@
           "alert_on_last_attempt": false
         },
         "webhook_notifications": {}
+      },
+      {
+        "task_key": "model_churn",
+        "depends_on": [
+          {
+            "task_key": "fs_dia_horario"
+          },
+          {
+            "task_key": "fs_geral"
+          },
+          {
+            "task_key": "fs_pontos"
+          },
+          {
+            "task_key": "fs_transacoes"
+          }
+        ],
+        "run_if": "ALL_SUCCESS",
+        "notebook_task": {
+          "notebook_path": "src/model_churn/predict",
+          "base_parameters": {
+            "date": "{{job.start_time.iso_date}}"
+          },
+          "source": "GIT"
+        },
+        "existing_cluster_id": "0809-155233-nc569ju7",
+        "timeout_seconds": 0,
+        "email_notifications": {}
       }
     ],
     "git_source": {