From 3573e37318b77df0f9258bdc3d01bca40bdebb21 Mon Sep 17 00:00:00 2001 From: andriisirak Date: Mon, 28 Jul 2025 17:45:55 +0200 Subject: [PATCH 1/3] Use AirFlow 2.10.5 --- requirements.txt | 8 ++++---- unaflow/dags/trigger_dag/trigger_dag_configuration.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/requirements.txt b/requirements.txt index 77ec43f..4c24d17 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ -apache-airflow==2.2.5 -apache-airflow-providers-google==6.7.0 +apache-airflow==2.10.5 +apache-airflow-providers-google==15.1.0 # Testing -pytest==5.3.5 +pytest==8.3.5 flake8==4.0.1 # Versioning and release versioneer==0.18 twine==5.0.0 -wheel==0.34.2 \ No newline at end of file +wheel==0.34.2 diff --git a/unaflow/dags/trigger_dag/trigger_dag_configuration.py b/unaflow/dags/trigger_dag/trigger_dag_configuration.py index d834dae..e40b0de 100644 --- a/unaflow/dags/trigger_dag/trigger_dag_configuration.py +++ b/unaflow/dags/trigger_dag/trigger_dag_configuration.py @@ -3,7 +3,7 @@ from typing import Dict, Optional, Sequence, Union from airflow.models.taskmixin import TaskMixin -from airflow.models.baseoperator import TaskStateChangeCallback +from airflow.models.abstractoperator import TaskStateChangeCallback from airflow.sensors.base import BaseSensorOperator from airflow.utils import timezone From e343d06a6a05c55e57e1769a46c1b87dfbe4f5ca Mon Sep 17 00:00:00 2001 From: andriisirak Date: Tue, 29 Jul 2025 13:19:40 +0200 Subject: [PATCH 2/3] updated workflow to use python version 3.11.9 --- .github/workflows/push.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml index 8d5d329..9632418 100644 --- a/.github/workflows/push.yml +++ b/.github/workflows/push.yml @@ -14,7 +14,7 @@ jobs: - name: Set-up python uses: actions/setup-python@v5 with: - python-version: 3.8 + python-version: 3.11.9 - name: Create cache id: cache @@ -34,4 +34,4 @@ jobs: - name: Make package run: | python setup.py sdist bdist_wheel - twine check dist/* \ No newline at end of file + twine check dist/* From 373c60178bc66b34133894b6763c06e131b11d81 Mon Sep 17 00:00:00 2001 From: andriisirak Date: Tue, 5 Aug 2025 14:20:59 +0200 Subject: [PATCH 3/3] Fix import for TaskStateChangeCallback to support Airflow versions < 2.7.0 --- unaflow/dags/trigger_dag/trigger_dag_configuration.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/unaflow/dags/trigger_dag/trigger_dag_configuration.py b/unaflow/dags/trigger_dag/trigger_dag_configuration.py index e40b0de..3d0f278 100644 --- a/unaflow/dags/trigger_dag/trigger_dag_configuration.py +++ b/unaflow/dags/trigger_dag/trigger_dag_configuration.py @@ -3,7 +3,13 @@ from typing import Dict, Optional, Sequence, Union from airflow.models.taskmixin import TaskMixin -from airflow.models.abstractoperator import TaskStateChangeCallback + +try: + from airflow.models.abstractoperator import TaskStateChangeCallback +except ImportError: + # For Airflow versions < 2.7.0 + from airflow.models.baseoperator import TaskStateChangeCallback + from airflow.sensors.base import BaseSensorOperator from airflow.utils import timezone