We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1518a9e commit 6193268Copy full SHA for 6193268
1 file changed
tests/hooks/dbt/test_dbt_hook_base.py
@@ -12,6 +12,7 @@
12
from airflow_dbt_python.hooks.fs.local import DbtLocalFsHook
13
from airflow_dbt_python.hooks.target import DbtConnectionHook, DbtPostgresHook
14
from airflow_dbt_python.utils.configs import RunTaskConfig
15
+from airflow_dbt_python.utils.version import AIRFLOW_V_3_0_PLUS
16
17
condition = False
18
try:
@@ -48,7 +49,10 @@ def test_dbt_hook_get_s3_remote():
48
49
50
def test_dbt_hook_get_local_fs_remote():
51
"""Test the correct remote is procured."""
- from airflow.hooks.filesystem import FSHook
52
+ if AIRFLOW_V_3_0_PLUS:
53
+ from airflow.providers.standard.hooks.filesystem import FSHook
54
+ else:
55
+ from airflow.hooks.filesystem import FSHook # type: ignore
56
57
hook = DbtHook()
58
0 commit comments