amoghrajesh commented on code in PR #53813:
URL: https://github.com/apache/airflow/pull/53813#discussion_r2236253903
##########
providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py:
##########
@@ -51,11 +50,12 @@
from airflow.utils.context import Context
if AIRFLOW_V_3_0_PLUS:
- from airflow.sdk import BaseOperatorLink
+ from airflow.sdk import BaseOperatorLink, TaskGroup
from airflow.sdk.execution_time.xcom import XCom
else:
from airflow.models import XCom
from airflow.models.baseoperatorlink import BaseOperatorLink # type:
ignore[no-redef]
+ from airflow.utils.task_group import TaskGroup # type: ignore[no-redef]
Review Comment:
Maybe we should replace these with try/except too
##########
providers/openlineage/tests/unit/openlineage/utils/test_utils.py:
##########
@@ -58,18 +58,18 @@
from airflow.timetables.trigger import CronTriggerTimetable
from airflow.utils import timezone
from airflow.utils.state import DagRunState
-from airflow.utils.task_group import TaskGroup
from airflow.utils.types import DagRunType
from tests_common.test_utils.compat import BashOperator, PythonOperator
from tests_common.test_utils.mock_operators import MockOperator
from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_3_PLUS,
AIRFLOW_V_3_0_PLUS
if AIRFLOW_V_3_0_PLUS:
- from airflow.sdk import BaseOperator, task
+ from airflow.sdk import BaseOperator, TaskGroup, task
else:
from airflow.decorators import task # type: ignore[attr-defined,no-redef]
from airflow.models.baseoperator import BaseOperator # type:
ignore[no-redef]
+ from airflow.utils.task_group import TaskGroup
Review Comment:
Same here
--
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]
