diff --git a/airflow/providers/google/cloud/transfers/gcs_to_gcs.py b/airflow/providers/google/cloud/transfers/gcs_to_gcs.py index cf4cf3c0ce657..d4c3bd679f700 100644 --- a/airflow/providers/google/cloud/transfers/gcs_to_gcs.py +++ b/airflow/providers/google/cloud/transfers/gcs_to_gcs.py @@ -553,7 +553,7 @@ def _copy_single_object(self, hook, source_object, destination_object): if self.move_object: hook.delete(self.source_bucket, source_object) - def get_openlineage_events_on_complete(self, task_instance): + def get_openlineage_facets_on_complete(self, task_instance): """ Implementing _on_complete because execute method does preprocessing on internals. This means we won't have to normalize self.source_object and self.source_objects, diff --git a/docs/apache-airflow-providers-openlineage/guides/developer.rst b/docs/apache-airflow-providers-openlineage/guides/developer.rst index 364f004830b40..4d5f7a5c67061 100644 --- a/docs/apache-airflow-providers-openlineage/guides/developer.rst +++ b/docs/apache-airflow-providers-openlineage/guides/developer.rst @@ -81,7 +81,7 @@ Here's example of properly implemented ``get_openlineage_facets_on_complete`` me .. code-block:: - def get_openlineage_events_on_complete(self, task_instance): + def get_openlineage_facets_on_complete(self, task_instance): """ Implementing _on_complete because execute method does preprocessing on internals. This means we won't have to normalize self.source_object and self.source_objects, diff --git a/tests/providers/google/cloud/transfers/test_gcs_to_gcs.py b/tests/providers/google/cloud/transfers/test_gcs_to_gcs.py index cf525235b14a1..f961038f5d4d8 100644 --- a/tests/providers/google/cloud/transfers/test_gcs_to_gcs.py +++ b/tests/providers/google/cloud/transfers/test_gcs_to_gcs.py @@ -840,7 +840,7 @@ def test_execute_simple_reports_openlineage(self, mock_hook): operator.execute(None) - lineage = operator.get_openlineage_events_on_complete(None) + lineage = operator.get_openlineage_facets_on_complete(None) assert len(lineage.inputs) == 1 assert len(lineage.outputs) == 1 assert lineage.inputs[0] == Dataset( @@ -862,7 +862,7 @@ def test_execute_multiple_reports_openlineage(self, mock_hook): operator.execute(None) - lineage = operator.get_openlineage_events_on_complete(None) + lineage = operator.get_openlineage_facets_on_complete(None) assert len(lineage.inputs) == 3 assert len(lineage.outputs) == 1 assert lineage.inputs == [ @@ -889,7 +889,7 @@ def test_execute_wildcard_reports_openlineage(self, mock_hook): operator.execute(None) - lineage = operator.get_openlineage_events_on_complete(None) + lineage = operator.get_openlineage_facets_on_complete(None) assert len(lineage.inputs) == 2 assert len(lineage.outputs) == 2 assert lineage.inputs == [