Remove Base class that not in use.

logging.basicConfig(level=logging.INFO)

LOGGER = logging.getLogger(__name__)

# Create a base class for declarative models

Base = registry().generate_base()

class AutomationModel(_Base):

    __tablename__ = 'automation'