# Import all models to ensure they are registered with SQLAlchemy
from models.TrainingProject import TrainingProject
from models.TrainingProjectDetails import TrainingProjectDetails
from models.training import Training
from models.LabelStudioProject import LabelStudioProject
from models.Images import Image
from models.Annotation import Annotation
from models.Settings import Settings
from models.ProjectClass import ProjectClass
from models.AnnotationProjectMapping import AnnotationProjectMapping
from models.ClassMapping import ClassMapping
from models.TrainingSize import TrainingSize

__all__ = [
    'TrainingProject',
    'TrainingProjectDetails',
    'Training',
    'LabelStudioProject',
    'Image',
    'Annotation',
    'Settings',
    'ProjectClass',
    'AnnotationProjectMapping',
    'ClassMapping',
    'TrainingSize'
]