Fix

astronomer · May 3, 2024 · 650957f · 650957f
1 parent 8a56b48
commit 650957f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/python-sdk/tests/airflow_tests/test_datasets.py b/python-sdk/tests/airflow_tests/test_datasets.py
@@ -104,7 +104,6 @@ def test_kwargs_with_temp_table():
 @pytest.mark.skipif(airflow.__version__ < "2.4.0", reason="Require Airflow version >= 2.4.0")
 def test_example_dataset_dag():
     from airflow.datasets import Dataset
-    from airflow.models.dataset import DatasetModel
 
     dir_path = os.path.dirname(os.path.realpath(__file__))
     db = DagBag(dir_path + "/../../example_dags/example_datasets.py")