From 4ac2cfc6139a636b6ecc19dc369d4f65e18eb174 Mon Sep 17 00:00:00 2001 From: David Betancur Date: Thu, 10 Oct 2024 16:00:16 -0400 Subject: [PATCH] Bugfix datatype error --- src/ingest-pipeline/airflow/dags/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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):