# 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

__all__ = [
    'TrainingProject',
    'TrainingProjectDetails',
    'Training',
    'LabelStudioProject',
    'Image',
    'Annotation'
]