corrige definição do daterange de rematerialização

prefeitura-rio · Apr 30, 2024 · afe68ef · afe68ef
1 parent a92998d
commit afe68ef
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pipelines/rj_smtr/br_rj_riodejaneiro_onibus_gps/flows.py b/pipelines/rj_smtr/br_rj_riodejaneiro_onibus_gps/flows.py
@@ -187,10 +187,12 @@
             name="assign_none_to_previous_runs",
         )()
     with case(rematerialization, True):
-        date_range_true = {
-            "date_range_start": date_range_start,
-            "date_range_end": date_range_end,
-        }
+        date_range_true = task(
+            lambda start, end: {
+                "date_range_start": start,
+                "date_range_end": end,
+            }
+        )(start=date_range_start, end=date_range_end)
         RUN_CLEAN_TRUE = clean_br_rj_riodejaneiro_onibus_gps(date_range_true)
 
     RUN_CLEAN = merge(RUN_CLEAN_TRUE, RUN_CLEAN_FALSE)