"""
Copyright (c) Meta Platforms, Inc. and affiliates.
"""

from . import kitti_trainer_ar, sintel_trainer_ar


def get_trainer(name):
    if name == "KITTI_AR":
        TrainFramework = kitti_trainer_ar.TrainFramework
    elif name == "SINTEL_AR":
        TrainFramework = sintel_trainer_ar.TrainFramework
    else:
        raise NotImplementedError(name)

    return TrainFramework