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 552d6efaffa18..71a58caf7f9a8 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.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 a9a2fb257ae99..1c43ca0eabffb 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.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 ed95bdb014192..5bdc3ce5db2a0 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 @@ # 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"