Move to airflow-pydantic

Author: timkpaine

airflow_config/__init__.py

@@ -1,5 +1,8 @@
 __version__ = "1.1.1"
 
+# Reexport
+from airflow_pydantic import BashOperatorArgs, Dag, DagArgs, PythonOperatorArgs, SSHOperatorArgs, Task, TaskArgs
+
 from .configuration import *
 from .dag import DAG, create_dag, create_dags, generate_dag_id
 from .exceptions import *

airflow_config/configuration/__init__.py

@@ -1,4 +1,2 @@
-from .airflow import Dag, DagArgs, Task, TaskArgs
 from .base import Configuration, load_config
-from .operators import *
 from .priority import *

airflow_config/configuration/airflow.py

@@ -3,11 +3,11 @@
 from pathlib import Path
 from typing import Dict, Optional
 
+from airflow_pydantic import Dag, DagArgs, Task, TaskArgs
 from hydra import compose, initialize_config_dir
 from hydra.utils import instantiate
 from pydantic import BaseModel, Field
 
-from airflow_config.configuration.airflow import Dag, DagArgs, Task, TaskArgs
 from airflow_config.exceptions import ConfigNotFoundError
 from airflow_config.utils import _get_calling_dag
 

airflow_config/configuration/base.py

@@ -29,6 +29,7 @@ classifiers = [
 ]
 
 dependencies = [
+    "airflow-pydantic",
     "apache-airflow>=2.8,<3",
     "colorlog",
     "hydra-core",