Fix Dataset import linting

apache · Oct 15, 2023 · e8d2159 · e8d2159
1 parent 92f0610
commit e8d2159
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/airflow/listeners/spec/dataset.py b/airflow/listeners/spec/dataset.py
@@ -22,7 +22,7 @@
 from pluggy import HookspecMarker
 
 if TYPE_CHECKING:
-    from airflow import Dataset
+    from airflow.datasets import Dataset
 
 hookspec = HookspecMarker("airflow")
 

diff --git a/tests/listeners/dataset_listener.py b/tests/listeners/dataset_listener.py
@@ -23,7 +23,7 @@
 from airflow.listeners import hookimpl
 
 if typing.TYPE_CHECKING:
-    from airflow.models.dataset import Dataset
+    from airflow.datasets import Dataset
 
 
 changed: list[Dataset] = []

diff --git a/tests/listeners/test_dataset_listener.py b/tests/listeners/test_dataset_listener.py
@@ -18,7 +18,7 @@
 
 import pytest
 
-from airflow import Dataset
+from airflow.datasets import Dataset
 from airflow.listeners.listener import get_listener_manager
 from airflow.models.dataset import DatasetModel
 from airflow.operators.empty import EmptyOperator