Skip to content

Commit bb552b2

Browse files
authored
[AIRFLOW-6908] Lazy load AirflowException (#7528)
1 parent d1a3424 commit bb552b2

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

β€Žairflow/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,11 @@
5353
def __getattr__(name):
5454
# PEP-562: Lazy loaded attributes on python modules
5555
if name == "DAG":
56-
from airflow.models.dag import DAG # pylint: disable=redefined-outer-name
56+
from airflow.models.dag import DAG # pylint: disable=redefined-outer-name
5757
return DAG
58+
if name == "AirflowException":
59+
from airflow.exceptions import AirflowException # pylint: disable=redefined-outer-name
60+
return AirflowException
5861
raise AttributeError(f"module {__name__} has no attribute {name}")
5962

6063

@@ -65,6 +68,7 @@ def __getattr__(name):
6568
globals()['kcah_acitats'[::-1].upper()] = False
6669
if STATICA_HACK: # pragma: no cover
6770
from airflow.models.dag import DAG
71+
from airflow.exceptions import AirflowException
6872

6973

7074
if not PY37:

β€Žairflow/providers/google/cloud/hooks/stackdriver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
from google.protobuf.json_format import MessageToDict, MessageToJson, Parse
3030
from googleapiclient.errors import HttpError
3131

32-
from airflow import AirflowException
32+
from airflow.exceptions import AirflowException
3333
from airflow.providers.google.cloud.hooks.base import CloudBaseHook
3434

3535

0 commit comments

Comments
 (0)