Fix formatting

datahub-project · Jun 18, 2024 · 424a4dc · 424a4dc
1 parent 62fdc92
commit 424a4dc
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...a-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/client/airflow_generator.py b/...a-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/client/airflow_generator.py
@@ -175,10 +175,11 @@ def generate_dataflow(
         data_flow.url = f"{base_url}/tree?dag_id={dag.dag_id}"
 
         if config.capture_ownership_info and dag.owner:
+            owners = [owner.strip() for owner in dag.owner.split(",")]
             if config.capture_ownership_as_group:
-                data_flow.group_owners.update(owner.strip() for owner in dag.owner.split(","))
+                data_flow.group_owners.update(owners)
             else:
-                data_flow.owners.update(owner.strip() for owner in dag.owner.split(","))
+                data_flow.owners.update(owners)
 
         if config.capture_tags_info and dag.tags:
             data_flow.tags.update(dag.tags)
@@ -281,7 +282,7 @@ def generate_datajob(
             datajob.url = f"{base_url}/taskinstance/list/?flt1_dag_id_equals={datajob.flow_urn.flow_id}&_flt_3_task_id={task.task_id}"
 
         if capture_owner and dag.owner:
-            if config.capture_ownership_as_group:
+            if config and config.capture_ownership_as_group:
                 datajob.group_owners.add(dag.owner)
             else:
                 datajob.owners.add(dag.owner)