metastable-projections/metastable_projections/projections/__init__.py

#TODO: License or such
from .base_projection_layer import BaseProjectionLayer
from .identity_projection_layer import IdentityProjectionLayer
from .frob_projection_layer import FrobeniusProjectionLayer
from .w2_projection_layer import WassersteinProjectionLayer

try:
    import cpp_projection
except ModuleNotFoundError:
    print('[MSB] ITPAL is not installed; KL projections not avaible.')
    from .base_projection_layer import ExceptionProjectionLayer as KLProjectionLayer
else:
    from .kl_projection_layer import KLProjectionLayer