fixing import error for dataset

apache · Jan 20, 2023 · d286a61 · d286a61
1 parent 10f0f8b
commit d286a61
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/airflow/__init__.py b/airflow/__init__.py
@@ -116,3 +116,4 @@ def __getattr__(name: str):
     from airflow.models.dag import DAG
     from airflow.models.xcom_arg import XComArg
     from airflow.exceptions import AirflowException
+    from airflow.models.dataset import Dataset