from .wasserstein import wasserstein_distance
from .bottleneck import bottleneck_distance