return_value

stellar · Apr 17, 2024 · 65def37 · 65def37
1 parent 9b85b70
commit 65def37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/stellar_etl_airflow/test_sources.py b/dags/stellar_etl_airflow/test_sources.py
@@ -29,7 +29,7 @@ def compare_transforms_and_bq_rows():
     for dag_run in execution_dates:
         # Retrieve successful_transforms from XCOM
         ti = TaskInstance(task, dag_run.execution_date)
-        xcom_ledgers = ti.xcom_pull(task_ids=task)
+        xcom_ledgers = ti.xcom_pull(task_ids=task, key="return_value")
 
         # Parse JSON and get successful_transforms
         successful_transforms_ledgers = json.loads(xcom_ledgers)[