diff --git a/src/ingest-pipeline/airflow/dags/utils.py b/src/ingest-pipeline/airflow/dags/utils.py index ec3a6951..4aaccb4d 100644 --- a/src/ingest-pipeline/airflow/dags/utils.py +++ b/src/ingest-pipeline/airflow/dags/utils.py @@ -326,7 +326,7 @@ def my_callable(**kwargs): def get_dataset_type_previous_version(**kwargs) -> List[str]: dataset_uuid = get_previous_revision_uuid(**kwargs) if dataset_uuid is None: - dataset_uuid = kwargs["dag_run"].conf.get("parent_submission_id", None) + dataset_uuid = kwargs["dag_run"].conf.get("parent_submission_id", [None])[0] assert dataset_uuid is not None, "Missing previous_version_uuid" def my_callable(**kwargs): @@ -340,7 +340,7 @@ def my_callable(**kwargs): def get_dataname_previous_version(**kwargs) -> str: dataset_uuid = get_previous_revision_uuid(**kwargs) if dataset_uuid is None: - dataset_uuid = kwargs["dag_run"].conf.get("parent_submission_id", None) + dataset_uuid = kwargs["dag_run"].conf.get("parent_submission_id", [None])[0] assert dataset_uuid is not None, "Missing previous_version_uuid" def my_callable(**kwargs):