diff --git a/airflow/chap9/tests/dags/test_dag_integrity.py b/airflow/chap9/tests/dags/test_dag_integrity.py index 53db518..8ed20d1 100644 --- a/airflow/chap9/tests/dags/test_dag_integrity.py +++ b/airflow/chap9/tests/dags/test_dag_integrity.py @@ -20,7 +20,7 @@ def test_dag_integrity(dag_file): module = importlib.util.module_from_spec(mod_spec) mod_spec.loader.exec_module(module) - dag_objects = [var for var in vars(module).values() + dag_objects = [var for var in vars(module).values() if isinstance(var, DAG)] assert dag_objects