diff --git a/dags/trigger_databricks_workflow.py b/dags/trigger_databricks_workflow.py index dc81a9e..06cee39 100644 --- a/dags/trigger_databricks_workflow.py +++ b/dags/trigger_databricks_workflow.py @@ -1,2 +1,15 @@ from airflow.sdk import DAG -from airflow.providers.databricks.operators \ No newline at end of file +from airflow.providers.databricks.operators.databricks import DatabricksRunNowOperator +from produce_data_assets import posts_asset, users_asset + +with DAG( + dag_id="trigger_databricks_workflow_dag", + schedule=(posts_asset & users_asset) +): + run_databricks_workflow = DatabricksRunNowOperator( + task_id="run_databricks_workflow", + databricks_conn_id="databricks_conn", + job_id="879157959924010" + ) + + run_databricks_workflow \ No newline at end of file