mudar forma de upstream

prefeitura-rio · Oct 18, 2023 · a33a4b8 · a33a4b8
1 parent 1644b72
commit a33a4b8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pipelines/rj_smtr/br_rj_riodejaneiro_bilhetagem/flows.py b/pipelines/rj_smtr/br_rj_riodejaneiro_bilhetagem/flows.py
@@ -146,9 +146,10 @@
         project_name=unmapped("staging"),
         parameters=constants.BILHETAGEM_CAPTURE_PARAMS.value,
         labels=unmapped(LABELS),
-        upstream_tasks=unmapped([wait_recaptura_trasacao]),
     )
 
+    runs_recaptura_auxiliar.set_upstream(wait_recaptura_trasacao)
+
     wait_recaptura_auxiliar = wait_for_flow_run.map(
         runs_recaptura_auxiliar,
         stream_states=unmapped(True),
@@ -163,9 +164,10 @@
         project_name=unmapped("staging"),
         parameters=constants.BILHETAGEM_CAPTURE_PARAMS.value,
         labels=unmapped(LABELS),
-        upstream_tasks=unmapped([wait_recaptura_auxiliar]),
     )
 
+    runs_captura.set_upstream(wait_recaptura_auxiliar)
+
     wait_captura = wait_for_flow_run.map(
         runs_captura,
         stream_states=unmapped(True),