File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
providers/google/cloud/hooks Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 53
53
def __getattr__ (name ):
54
54
# PEP-562: Lazy loaded attributes on python modules
55
55
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
57
57
return DAG
58
+ if name == "AirflowException" :
59
+ from airflow .exceptions import AirflowException # pylint: disable=redefined-outer-name
60
+ return AirflowException
58
61
raise AttributeError (f"module { __name__ } has no attribute { name } " )
59
62
60
63
@@ -65,6 +68,7 @@ def __getattr__(name):
65
68
globals ()['kcah_acitats' [::- 1 ].upper ()] = False
66
69
if STATICA_HACK : # pragma: no cover
67
70
from airflow .models .dag import DAG
71
+ from airflow .exceptions import AirflowException
68
72
69
73
70
74
if not PY37 :
Original file line number Diff line number Diff line change 29
29
from google .protobuf .json_format import MessageToDict , MessageToJson , Parse
30
30
from googleapiclient .errors import HttpError
31
31
32
- from airflow import AirflowException
32
+ from airflow . exceptions import AirflowException
33
33
from airflow .providers .google .cloud .hooks .base import CloudBaseHook
34
34
35
35
You canβt perform that action at this time.
0 commit comments