From dd23f261b01252df755974fedef5fa75239ebe47 Mon Sep 17 00:00:00 2001 From: Kacper Muda Date: Wed, 1 Jul 2026 09:29:37 +0200 Subject: [PATCH] Fix hll import in openlineage system tests --- .../example_openlineage_policy_dag_events_false_dag.py | 2 +- .../example_openlineage_policy_hook_lineage_dag.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py index 00342319c52e4..52d754b3132e2 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py @@ -35,7 +35,7 @@ def _register_hook_lineage(): - from airflow.lineage.hook import get_hook_lineage_collector + from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector collector = get_hook_lineage_collector() collector.add_input_asset(context=None, uri="file://host1/in1.txt") diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py index 721672808b85d..6f1e8a76f972f 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py @@ -37,7 +37,7 @@ def _register_hook_lineage(): - from airflow.lineage.hook import get_hook_lineage_collector + from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector collector = get_hook_lineage_collector() collector.add_input_asset(context=None, uri="file://host1/in1.txt") @@ -47,7 +47,7 @@ def _register_hook_lineage(): def _register_sql_hook_lineage(): - from airflow.lineage.hook import get_hook_lineage_collector + from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector from airflow.providers.common.sql.hooks.lineage import SqlJobHookLineageExtra collector = get_hook_lineage_collector()