from .information_bottleneck_comm import IBComm
from .information_bottleneck_comm_full import IBFComm
from .information_bottleneck_comm_not_IB import IBNIBComm
from .tar_comm import TarComm
from .information_bottleneck_pruned_comm import IBPComm
from .id_comm import IdComm
from .orc_comm import OrcComm
from .new_ib import NewIB
from .information_bottleneck_comm_full_simi import IBFCommS
from .information_bottleneck_comm_full_mle import IBFCommM
from .information_bottleneck_comm_full_mlev27 import IBFCommM27
from .information_bottleneck_comm_full_similv import IBFCommSV
from .information_bottleneck_comm_full_simipca import IBFCommSP

REGISTRY = {"information_bottleneck": IBComm,
            "information_bottleneck_full": IBFComm,
            "information_bottleneck_full_simi": IBFCommS,
            "information_bottleneck_full_mle": IBFCommM,
            "information_bottleneck_full_mlev27": IBFCommM27,
            "information_bottleneck_full_similv": IBFCommSV,
            "information_bottleneck_full_simipca": IBFCommSP,
            "information_bottleneck_not_IB": IBNIBComm,
            "tar": TarComm,
            "information_bottleneck_pruned": IBPComm,
            "id_comm": IdComm,
            "orc_comm": OrcComm,
            "new_ib": NewIB}
