This is an automated email from the ASF dual-hosted git repository.
amoghdesai 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 3b4df0cd3d4 Use Label class from task sdk in providers (#49398)
3b4df0cd3d4 is described below
commit 3b4df0cd3d4920e09de8f5e66cc1d75e6c22e3ae
Author: Amogh Desai <amoghrajesh1...@gmail.com>
AuthorDate: Thu Apr 17 17:00:01 2025 +0530
Use Label class from task sdk in providers (#49398)
---
.../tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py | 2 +-
providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py | 2 +-
.../azure/tests/system/microsoft/azure/example_adf_run_pipeline.py | 2 +-
3 files changed, 3 insertions(+), 3 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 552d6efaffa..71a58caf7f9 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
@@ -58,7 +58,7 @@ from
airflow.providers.amazon.aws.sensors.opensearch_serverless import (
)
from airflow.providers.amazon.aws.utils import get_botocore_version
from airflow.providers.standard.operators.empty import EmptyOperator
-from airflow.utils.edgemodifier import Label
+from airflow.sdk import Label
from airflow.utils.trigger_rule import TriggerRule
from system.amazon.aws.utils import SystemTestContextBuilder
diff --git a/providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py
b/providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py
index a9a2fb257ae..1c43ca0eabf 100644
--- a/providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py
+++ b/providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py
@@ -26,7 +26,7 @@ from airflow.providers.dbt.cloud.operators.dbt import (
)
from airflow.providers.dbt.cloud.sensors.dbt import DbtCloudJobRunSensor
from airflow.providers.standard.operators.empty import EmptyOperator
-from airflow.utils.edgemodifier import Label
+from airflow.sdk import Label
from tests_common.test_utils.system_tests import get_test_env_id
diff --git
a/providers/microsoft/azure/tests/system/microsoft/azure/example_adf_run_pipeline.py
b/providers/microsoft/azure/tests/system/microsoft/azure/example_adf_run_pipeline.py
index ed95bdb0141..5bdc3ce5db2 100644
---
a/providers/microsoft/azure/tests/system/microsoft/azure/example_adf_run_pipeline.py
+++
b/providers/microsoft/azure/tests/system/microsoft/azure/example_adf_run_pipeline.py
@@ -28,7 +28,7 @@ from airflow.providers.microsoft.azure.sensors.data_factory
import AzureDataFact
# Ignore missing args provided by default_args
# mypy: disable-error-code="call-arg"
from airflow.providers.standard.operators.empty import EmptyOperator
-from airflow.utils.edgemodifier import Label
+from airflow.sdk import Label
ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
DAG_ID = "example_adf_run_pipeline"
