Use AirFlow 2.10.5

.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/*
+          twine check dist/*

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
+wheel==0.34.2

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.baseoperator 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