from ml_lib.utils.config import Config from models.binary_classifier import BinaryClassifier class MConfig(Config): @property def _model_map(self): return dict(BinaryClassifier=BinaryClassifier)