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] Support remote workspace for RunJobTask #140

Closed maxim-mityutko closed 1 month ago

maxim-mityutko commented 1 month ago

Description

Databricks does not natively support riggering the job run in the remote workspace. This feature implements the logic that either creates a native RunJobTask or leverages the new RunJobInRemoteWorkspace plugin to execute both local and remote workspace job runs. The functionality does not require any additional parameters, and leverage the host parameter of the RunJobTask with the assumtion that if host is provided, user wants remote run. If / when Databricks implements this functionality natively, the code can be removed, but existing workflows won't require any changes. If host is not provided, Databricks SDK will retrieve host and token for the DEFAULT profile.

Related Issue

139

Motivation and Context

How Has This Been Tested?

Unit tests, custom workflow image image

Screenshots (if appropriate):

Types of changes

Checklist:

maxim-mityutko commented 1 month ago

@asingamaneni please let me know what you think about this approach