uranusjr commented on code in PR #55110:
URL: https://github.com/apache/airflow/pull/55110#discussion_r2430920267
##########
providers/google/src/airflow/providers/google/cloud/operators/workflows.py:
##########
@@ -36,12 +37,16 @@
)
from airflow.providers.google.cloud.operators.cloud_base import
GoogleCloudBaseOperator
from airflow.providers.google.common.hooks.base_google import
PROVIDE_PROJECT_ID
+from airflow.providers.google.version_compat import AIRFLOW_V_3_0_PLUS
if TYPE_CHECKING:
from google.api_core.retry import Retry
from google.protobuf.field_mask_pb2 import FieldMask
- from airflow.utils.context import Context
+ if AIRFLOW_V_3_0_PLUS:
+ from airflow.sdk.definitions.context import Context
+ else:
+ from airflow.utils.context import Context
Review Comment:
Same as above
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
