This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 2ff8a72572a Fix various Airflow 2 import issues revealed by previous
fixes (#59341)
2ff8a72572a is described below
commit 2ff8a72572a40a4d5a6b9b6cc27b8bf4d740660b
Author: Niko Oliveira <[email protected]>
AuthorDate: Fri Dec 12 13:12:51 2025 -0800
Fix various Airflow 2 import issues revealed by previous fixes (#59341)
---
.../tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py | 2 +-
providers/amazon/tests/system/amazon/aws/example_mwaa.py | 1 +
providers/amazon/tests/system/amazon/aws/example_sagemaker.py | 2 +-
3 files changed, 3 insertions(+), 2 deletions(-)
diff --git
a/providers/amazon/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
b/providers/amazon/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
index 3e0d2a84868..dd0c543968a 100644
---
a/providers/amazon/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
+++
b/providers/amazon/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py
@@ -66,7 +66,7 @@ else:
from airflow.decorators import task, task_group # type:
ignore[attr-defined,no-redef]
from airflow.models.baseoperator import chain # type:
ignore[attr-defined,no-redef]
from airflow.models.dag import DAG # type:
ignore[attr-defined,no-redef,assignment]
- from airflow.sdk import Label # type: ignore[attr-defined,no-redef]
+ from airflow.utils.edgemodifier import Label # type:
ignore[attr-defined,no-redef]
try:
from airflow.sdk import TriggerRule
except ImportError:
diff --git a/providers/amazon/tests/system/amazon/aws/example_mwaa.py
b/providers/amazon/tests/system/amazon/aws/example_mwaa.py
index 8dbcc626159..454fe192c49 100644
--- a/providers/amazon/tests/system/amazon/aws/example_mwaa.py
+++ b/providers/amazon/tests/system/amazon/aws/example_mwaa.py
@@ -34,6 +34,7 @@ else:
from airflow.decorators import task # type: ignore[attr-defined,no-redef]
from airflow.models.baseoperator import chain # type:
ignore[attr-defined,no-redef]
from airflow.models.dag import DAG # type:
ignore[attr-defined,no-redef,assignment]
+ from airflow.utils import timezone # type: ignore[attr-defined,no-redef]
from system.amazon.aws.utils import SystemTestContextBuilder
diff --git a/providers/amazon/tests/system/amazon/aws/example_sagemaker.py
b/providers/amazon/tests/system/amazon/aws/example_sagemaker.py
index d2a5e08fcde..c6eed2a3297 100644
--- a/providers/amazon/tests/system/amazon/aws/example_sagemaker.py
+++ b/providers/amazon/tests/system/amazon/aws/example_sagemaker.py
@@ -436,7 +436,7 @@ def set_up(env_id, role_arn):
if AIRFLOW_V_3_0_PLUS:
from airflow.sdk import get_current_context
else:
- from airflow.providers.standard.operators.python import
get_current_context
+ from airflow.operators.python import get_current_context # type:
ignore[attr-defined,no-redef]
ti = get_current_context()["ti"]
ti.xcom_push(key="docker_image", value=ecr_repository_uri)
