Fix tests

apache · Feb 21, 2024 · bacd50f · bacd50f
1 parent 801ae6b
commit bacd50f
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py b/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py
@@ -22,13 +22,6 @@
 
 from flask import session, url_for
 
-from airflow.auth.managers.models.resource_details import (
-    ConnectionDetails,
-    DagAccessEntity,
-    DagDetails,
-    PoolDetails,
-    VariableDetails,
-)
 from airflow.cli.cli_config import CLICommand, DefaultHelpParser, GroupCommand
 from airflow.configuration import conf
 from airflow.exceptions import AirflowOptionalProviderFeatureException
@@ -50,6 +43,13 @@
 
 try:
     from airflow.auth.managers.base_auth_manager import BaseAuthManager, ResourceMethod
+    from airflow.auth.managers.models.resource_details import (
+        ConnectionDetails,
+        DagAccessEntity,
+        DagDetails,
+        PoolDetails,
+        VariableDetails,
+    )
 except ImportError:
     raise AirflowOptionalProviderFeatureException(
         "Failed to import BaseUser. This feature is only available in Airflow versions >= 2.8.0"