from .contrastive import Contrastive
from .w_mse import WMSE
from .byol import BYOL


METHOD_LIST = ["contrastive", "w_mse", "byol"]


def get_method(name):
    assert name in METHOD_LIST
    if name == "contrastive":
        return Contrastive
    elif name == "w_mse":
        return WMSE
    elif name == "byol":
        return BYOL

from .contrastive import Contrastive
from .w_mse import WMSE
from .byol import BYOL


METHOD_LIST = ["contrastive", "w_mse", "byol"]


def get_method(name):
    assert name in METHOD_LIST
    if name == "contrastive":
        return Contrastive
    elif name == "w_mse":
        return WMSE
    elif name == "byol":
        return BYOL
