Skip to content

Commit 411de03

Browse files
committed
Fix KPO import
1 parent 6dd9383 commit 411de03

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

dagfactory/dagbuilder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@
7070
logger.info("Package apache-airflow-providers-common-sql is not installed.")
7171
SQL_SENSOR_CLASS = None
7272

73+
# Try to import KubernetesPodOperator only if the package is installed
7374
try:
7475
from airflow.providers.cncf.kubernetes.operators.pod import KubernetesPodOperator
7576

@@ -468,7 +469,8 @@ def make_task(operator: str, task_params: Dict[str, Any]) -> BaseOperator:
468469
):
469470
task_params = DagBuilder._handle_http_sensor(operator_obj, task_params)
470471

471-
if KUBERNETES_OPERATOR_CLASS and issubclass(operator_obj, KubernetesPodOperator):
472+
# Only handle KubernetesPodOperator if the package is installed
473+
if KUBERNETES_OPERATOR_CLASS and issubclass(operator_obj, KUBERNETES_OPERATOR_CLASS):
472474
task_params = DagBuilder._clean_kpo_task_params(task_params)
473475

474476
DagBuilder.adjust_general_task_params(task_params)

0 commit comments

Comments
 (0)