#1115 basehook fix

CityofToronto · Dec 17, 2024 · 4a63acc · 4a63acc
1 parent be06be0
commit 4a63acc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/volumes/miovision/api/intersection_tmc.py b/volumes/miovision/api/intersection_tmc.py
@@ -12,7 +12,7 @@
 from time import sleep
 from collections import namedtuple
 
-from airflow.hooks.base_hook import BaseHook
+from airflow.hooks.base import BaseHook
 from airflow.providers.postgres.hooks.postgres import PostgresHook
 
 class BreakingError(Exception):

diff --git a/volumes/miovision/api/pull_alert.py b/volumes/miovision/api/pull_alert.py
@@ -11,7 +11,7 @@
 from psycopg2 import sql
 from psycopg2.extras import execute_values
 
-from airflow.hooks.base_hook import BaseHook
+from airflow.hooks.base import BaseHook
 from airflow.providers.postgres.hooks.postgres import PostgresHook
 
 def logger():