from ml_lib.utils.config import Config
from models.binary_classifier import BinaryClassifier
from models.bandwise_binary_classifier import BandwiseBinaryClassifier


class MConfig(Config):
    # TODO: There should be a way to automate this.

    @property
    def _model_map(self):
        return dict(BinaryClassifier=BinaryClassifier,
                    BandwiseBinaryClassifier=BandwiseBinaryClassifier)