From 7a4b0d19570cb75e203e083566a48f850dd7164d Mon Sep 17 00:00:00 2001
From: Waleed Akbar <wakbar@cttc.es>
Date: Fri, 7 Jun 2024 09:32:48 +0000
Subject: [PATCH] import of Base class of updated to newer version

---
 src/kpi_manager/service/database/KpiModel.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/kpi_manager/service/database/KpiModel.py b/src/kpi_manager/service/database/KpiModel.py
index 16844fdc0..9ab98e4ef 100644
--- a/src/kpi_manager/service/database/KpiModel.py
+++ b/src/kpi_manager/service/database/KpiModel.py
@@ -15,7 +15,8 @@
 import logging
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy import Column, Integer, String, Float, Text, ForeignKey
-from sqlalchemy.ext.declarative import declarative_base
+# from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import registry
 from sqlalchemy.orm import sessionmaker, relationship
 
 
@@ -23,7 +24,8 @@ logging.basicConfig(level=logging.INFO)
 LOGGER = logging.getLogger(__name__)
 
 # Create a base class for declarative models
-Base = declarative_base()
+Base = registry().generate_base()
+# Base = declarative_base()
 
 class Kpi(Base):
     __tablename__ = 'kpi'
-- 
GitLab