From 4a63accdd9a76b831e31307e9af947c8b4c7c3a0 Mon Sep 17 00:00:00 2001
From: gabrielwol <80077912+gabrielwol@users.noreply.github.com>
Date: Tue, 17 Dec 2024 18:55:53 +0000
Subject: [PATCH] #1115 basehook fix

---
 volumes/miovision/api/intersection_tmc.py | 2 +-
 volumes/miovision/api/pull_alert.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/volumes/miovision/api/intersection_tmc.py b/volumes/miovision/api/intersection_tmc.py
index ce41fa022..997639ace 100644
--- 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
index e07685ed9..555b2e955 100644
--- 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():