import of Base class of updated to newer version

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

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'