from federatedscope.register import register_trainer
from federatedscope.core.trainers.torch_trainer import GeneralTorchTrainer


# Build your trainer here.
class MyTrainer(GeneralTorchTrainer):
    pass


def call_my_trainer(trainer_type):
    if trainer_type == 'mytrainer':
        trainer_builder = MyTrainer
        return trainer_builder


register_trainer('mytrainer', call_my_trainer)