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/* 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..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.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