Skip to content

Commit 3b4df0c

Browse files
authored
Use Label class from task sdk in providers (apache#49398)
1 parent 4772292 commit 3b4df0c

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

providers/amazon/tests/system/amazon/aws/example_bedrock_retrieve_and_generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
)
5959
from airflow.providers.amazon.aws.utils import get_botocore_version
6060
from airflow.providers.standard.operators.empty import EmptyOperator
61-
from airflow.utils.edgemodifier import Label
61+
from airflow.sdk import Label
6262
from airflow.utils.trigger_rule import TriggerRule
6363

6464
from system.amazon.aws.utils import SystemTestContextBuilder

providers/dbt/cloud/tests/system/dbt/cloud/example_dbt_cloud.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
)
2727
from airflow.providers.dbt.cloud.sensors.dbt import DbtCloudJobRunSensor
2828
from airflow.providers.standard.operators.empty import EmptyOperator
29-
from airflow.utils.edgemodifier import Label
29+
from airflow.sdk import Label
3030

3131
from tests_common.test_utils.system_tests import get_test_env_id
3232

providers/microsoft/azure/tests/system/microsoft/azure/example_adf_run_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
# Ignore missing args provided by default_args
2929
# mypy: disable-error-code="call-arg"
3030
from airflow.providers.standard.operators.empty import EmptyOperator
31-
from airflow.utils.edgemodifier import Label
31+
from airflow.sdk import Label
3232

3333
ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
3434
DAG_ID = "example_adf_run_pipeline"

0 commit comments

Comments
 (0)