From a03c9bc17b1a9820c9b3e55872ed5be142ead447 Mon Sep 17 00:00:00 2001 From: Marko Djordjevic Date: Wed, 18 Feb 2026 20:57:37 +0100 Subject: [PATCH] code-review-fix task 15.2: replace declarative_base() with class Base(DeclarativeBase) in db.py --- services/ml/app/db.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/services/ml/app/db.py b/services/ml/app/db.py index 13811b7..8fc8475 100644 --- a/services/ml/app/db.py +++ b/services/ml/app/db.py @@ -10,8 +10,7 @@ from contextlib import contextmanager from typing import Generator from sqlalchemy import create_engine, Column, Integer, String, DateTime, JSON -from sqlalchemy.ext.declarative import declarative_base -from sqlalchemy.orm import sessionmaker, Session +from sqlalchemy.orm import DeclarativeBase, sessionmaker, Session from sqlalchemy.sql import func @@ -35,7 +34,8 @@ engine = create_engine( SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) # Base class for declarative models -Base = declarative_base() +class Base(DeclarativeBase): + pass # Training runs model