Simplify expression for get_permitted_dag_ids query (apache#42484)

aws-mwaa · Sep 26, 2024 · 2beb6a7 · 2beb6a7
1 parent 8e56206
commit 2beb6a7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/airflow/providers/fab/auth_manager/fab_auth_manager.py b/airflow/providers/fab/auth_manager/fab_auth_manager.py
@@ -342,10 +342,7 @@ def get_permitted_dag_ids(
                         resources.add(resource[len(permissions.RESOURCE_DAG_PREFIX) :])
                     else:
                         resources.add(resource)
-        return {
-            dag.dag_id
-            for dag in session.execute(select(DagModel.dag_id).where(DagModel.dag_id.in_(resources)))
-        }
+        return set(session.scalars(select(DagModel.dag_id).where(DagModel.dag_id.in_(resources))))
 
     @cached_property
     def security_manager(self) -> FabAirflowSecurityManagerOverride: