src/abstract_trainer.py

# -------------------------------------------------------------------
# Pimcore
#
# This source file is available under two different licenses:
#  - GNU General Public License version 3 (GPLv3)
#  - Pimcore Commercial License (PCL)
# Full copyright and license information is available in
# LICENSE.md which is distributed with this source code.
#
#  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
#  @license    http://www.pimcore.org/license     GPLv3 and PCL
# -------------------------------------------------------------------


from abc import ABC, abstractmethod

import logging
from .training_status import TrainingStatus

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)


class AbstractTrainer(ABC): 

    __training_status: TrainingStatus = TrainingStatus(); 

    @abstractmethod
    async def start_training(self):
        logger.info('start abstract trainer training')
        pass

    def get_status(self) -> TrainingStatus:
        return self.__training_status