comparing probability distributions with Wasserstein distance