Nike-Inc / brickflow

Pythonic Programming Framework to orchestrate jobs in Databricks Workflow
https://engineering.nike.com/brickflow/
Apache License 2.0
183 stars 36 forks source link

[FEATURE] Add `WorkflowTaskDependencySensor` #110

Closed maxim-mityutko closed 3 months ago

maxim-mityutko commented 4 months ago

Description

Leverage Databricks API to retrieve the state of the specific task within the workflow run. The code mostly inherits from the WorkflowDependencySensor.

Related Issue

109

Motivation and Context

In many cases it's more benificial to put a dependency on a specific task within the workflow instead the workflow itself. It reduces the wait time of the downstream workflows.

How Has This Been Tested?

Unit tests + workflow in Databricks with a success sensor and a failed sensor

Screenshots (if appropriate):

Types of changes

Checklist: