Skip to content

Commit 1dd694e

Browse files
authored
fix: fix depends_on_past issue for sensors (#763)
1 parent 70a84c3 commit 1dd694e

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

Diff for: ext/scheduler/airflow2/resources/base_dag.py

+6
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@
7373
task_id = JOB_START_EVENT_NAME,
7474
python_callable = log_job_start,
7575
provide_context=True,
76+
depends_on_past=False,
7677
dag=dag
7778
)
7879

@@ -81,6 +82,7 @@
8182
python_callable = log_job_end,
8283
provide_context=True,
8384
trigger_rule= 'all_success',
85+
depends_on_past=False,
8486
dag=dag
8587
)
8688

@@ -219,6 +221,7 @@
219221
get_logs=True,
220222
dag=dag,
221223
in_cluster=True,
224+
depends_on_past=False,
222225
is_delete_operator_pod=True,
223226
do_xcom_push=False,
224227
env_vars=executor_env_vars,
@@ -252,6 +255,7 @@
252255
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
253256
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
254257
task_id="wait_{{$dependency.Job.Name | trunc 200}}-{{$dependencySchema.Name}}",
258+
depends_on_past=False,
255259
dag=dag
256260
)
257261
{{- end}}
@@ -268,6 +272,7 @@
268272
window_version=int("{{ $baseWindow.GetVersion }}"),
269273
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
270274
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
275+
depends_on_past=False,
271276
task_id="wait_{{$dependency.JobName | trunc 200}}-{{$dependency.TaskName}}",
272277
dag=dag
273278
)
@@ -284,6 +289,7 @@
284289
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
285290
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
286291
task_id='wait_{{$httpDependency.Name| trunc 200}}',
292+
depends_on_past=False,
287293
dag=dag
288294
)
289295
{{- end}}

Diff for: ext/scheduler/airflow2/resources/expected_compiled_template.py

+8
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@
6565
task_id = JOB_START_EVENT_NAME,
6666
python_callable = log_job_start,
6767
provide_context=True,
68+
depends_on_past=False,
6869
dag=dag
6970
)
7071

@@ -73,6 +74,7 @@
7374
python_callable = log_job_end,
7475
provide_context=True,
7576
trigger_rule= 'all_success',
77+
depends_on_past=False,
7678
dag=dag
7779
)
7880

@@ -174,6 +176,7 @@
174176
get_logs=True,
175177
dag=dag,
176178
in_cluster=True,
179+
depends_on_past=False,
177180
is_delete_operator_pod=True,
178181
do_xcom_push=False,
179182
env_vars=executor_env_vars,
@@ -211,6 +214,7 @@
211214
get_logs=True,
212215
dag=dag,
213216
in_cluster=True,
217+
depends_on_past=False,
214218
is_delete_operator_pod=True,
215219
do_xcom_push=False,
216220
env_vars=executor_env_vars,
@@ -248,6 +252,7 @@
248252
get_logs=True,
249253
dag=dag,
250254
in_cluster=True,
255+
depends_on_past=False,
251256
is_delete_operator_pod=True,
252257
do_xcom_push=False,
253258
env_vars=executor_env_vars,
@@ -273,6 +278,7 @@
273278
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
274279
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
275280
task_id="wait_foo-intra-dep-job-bq",
281+
depends_on_past=False,
276282
dag=dag
277283
)
278284
wait_foo__dash__inter__dash__dep__dash__job = SuperExternalTaskSensor(
@@ -286,6 +292,7 @@
286292
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
287293
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
288294
task_id="wait_foo-inter-dep-job-bq",
295+
depends_on_past=False,
289296
dag=dag
290297
)
291298

@@ -299,6 +306,7 @@
299306
window_version=int("1"),
300307
poke_interval=SENSOR_DEFAULT_POKE_INTERVAL_IN_SECS,
301308
timeout=SENSOR_DEFAULT_TIMEOUT_IN_SECS,
309+
depends_on_past=False,
302310
task_id="wait_foo-external-optimus-dep-job-bq",
303311
dag=dag
304312
)

0 commit comments

Comments
 (0)