Skip to content

Commit 55706c4

Browse files
committed
Fix tests
1 parent 411de03 commit 55706c4

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

dagfactory/dagbuilder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def _clean_kpo_task_params(task_params: dict) -> dict:
354354
@staticmethod
355355
def _handle_http_sensor(operator_obj, task_params):
356356
# Only handle if HttpOperator/HttpSensor are available
357-
if HTTP_OPERATOR_CLASS and isinstance(operator_obj, HTTP_OPERATOR_CLASS):
357+
if HTTP_OPERATOR_CLASS and issubclass(operator_obj, HTTP_OPERATOR_CLASS):
358358
headers = task_params.get("headers", {})
359359
content_type = headers.get("Content-Type", "").lower()
360360

@@ -388,7 +388,7 @@ def _handle_http_sensor(operator_obj, task_params):
388388
# remove dag-factory specific parameters
389389
# Airflow 2.0 doesn't allow these to be passed to operator
390390
del task_params["response_check_lambda"]
391-
return task_params
391+
return task_params
392392

393393
# pylint: disable=too-many-branches
394394
# pylint: disable=too-many-statements

scripts/test/pre-install-airflow.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@ if [ "$AIRFLOW_VERSION" = "3.0" ]; then
3636
uv pip install "apache-airflow>=3.0.2" --constraint /tmp/constraint.txt
3737
else
3838
uv pip install "apache-airflow==$AIRFLOW_VERSION" --constraint /tmp/constraint.txt
39-
uv pip install apache-airflow-providers-cncf-kubernetes --constraint /tmp/constraint.txt
4039
fi;
4140

4241
rm /tmp/constraint.txt

0 commit comments

Comments
 (0)