ch05 - editor comments

chapter05/dags/02_branch_function.py

@@ -27,7 +27,7 @@ def _fetch_sales_new(**context):
 
 
 def _clean_sales(**context):
-    if context["execution_date"] < pendulum.today("UTC").add(days=-1):
+    if context["data_interval_start"] < ERP_CHANGE_DATE:
         _clean_sales_old(**context)
     else:
         _clean_sales_new(**context)

chapter05/dags/05_condition_function.py

@@ -13,7 +13,7 @@
 
 
 def _pick_erp_system(**context):
-    if context["data_interval_start"] < pendulum.today("UTC").add(days=-1):
+    if context["data_interval_start"] < ERP_CHANGE_DATE:
         return "fetch_sales_old"
     else:
         return "fetch_sales_new"

chapter05/dags/13_dag_decorator.py

@@ -14,7 +14,7 @@
 )
 def taskflow_api_decorator(): 
 
-    @task #C
+    @task 
     def train_model():
         model_id = str(uuid.uuid4())
         return model_id