diff --git a/providers/src/airflow/providers/apache/spark/hooks/spark_submit.py b/providers/src/airflow/providers/apache/spark/hooks/spark_submit.py index 1aab2dc0ea751..9889cf9473508 100644 --- a/providers/src/airflow/providers/apache/spark/hooks/spark_submit.py +++ b/providers/src/airflow/providers/apache/spark/hooks/spark_submit.py @@ -519,14 +519,14 @@ def _build_track_driver_status_command(self) -> list[str]: def _resolve_kerberos_principal(self, principal: str | None) -> str: """Resolve kerberos principal.""" # todo: remove try/exception when min airflow version is 3.0 - try: - from airflow.security.kerberos import get_kerberos_principal # type: ignore[attr-defined] - except ImportError: - from airflow.security.kerberos import ( - get_kerberos_principle as get_kerberos_principal, # type: ignore[attr-defined] - ) + from airflow.security import kerberos - return get_kerberos_principal(principal) + try: + func = kerberos.get_kerberos_principal + except AttributeError: + # Fallback for older versions of Airflow + func = kerberos.get_kerberos_principle # type: ignore[attr-defined] + return func(principal) def submit(self, application: str = "", **kwargs: Any) -> None: """