openlineage: remove deprecated parentRun facet key (#40681)

Signed-off-by: Kacper Muda <[email protected]>
apache · Jul 10, 2024 · 14b075b · 14b075b
1 parent 22ec726
commit 14b075b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 18 deletions.
diff --git a/airflow/providers/openlineage/plugins/adapter.py b/airflow/providers/openlineage/plugins/adapter.py
@@ -434,12 +434,7 @@ def _build_run(
                 namespace=conf.namespace(),
                 name=parent_job_name or job_name,
             )
-            facets.update(
-                {
-                    "parent": parent_run_facet,
-                    "parentRun": parent_run_facet,  # Keep sending this for the backward compatibility
-                }
-            )
+            facets.update({"parent": parent_run_facet})
 
         if run_facets:
             facets.update(run_facets)

diff --git a/tests/providers/openlineage/plugins/test_adapter.py b/tests/providers/openlineage/plugins/test_adapter.py
@@ -243,10 +243,6 @@ def test_emit_start_event_with_additional_information(mock_stats_incr, mock_stat
                             run={"runId": "parent_run_id"},
                             job={"namespace": namespace(), "name": "parent_job_name"},
                         ),
-                        "parentRun": ParentRunFacet(
-                            run={"runId": "parent_run_id"},
-                            job={"namespace": namespace(), "name": "parent_job_name"},
-                        ),
                         "externalQuery1": ExternalQueryRunFacet(externalQueryId="123", source="source"),
                         "externalQuery2": ExternalQueryRunFacet(externalQueryId="999", source="source"),
                     },
@@ -361,10 +357,6 @@ def test_emit_complete_event_with_additional_information(mock_stats_incr, mock_s
                             run={"runId": "parent_run_id"},
                             job={"namespace": namespace(), "name": "parent_job_name"},
                         ),
-                        "parentRun": ParentRunFacet(
-                            run={"runId": "parent_run_id"},
-                            job={"namespace": namespace(), "name": "parent_job_name"},
-                        ),
                         "externalQuery": ExternalQueryRunFacet(externalQueryId="123", source="source"),
                     },
                 ),
@@ -472,10 +464,6 @@ def test_emit_failed_event_with_additional_information(mock_stats_incr, mock_sta
                             run={"runId": "parent_run_id"},
                             job={"namespace": namespace(), "name": "parent_job_name"},
                         ),
-                        "parentRun": ParentRunFacet(
-                            run={"runId": "parent_run_id"},
-                            job={"namespace": namespace(), "name": "parent_job_name"},
-                        ),
                         "externalQuery": ExternalQueryRunFacet(externalQueryId="123", source="source"),
                         "errorMessage": ErrorMessageRunFacet(
                             message="Error message", programmingLanguage="python", stackTrace=None