Wasserstein_metric loading ...