For two probability distributions $mu$ and $nu$ defined on $X$, the $p$-th Wasserstein distance between the two of them is defined as $$W_p(mu,nu) = left(inf_{piinPi(mu,nu)}int_{Xtimes X}c(x,x’)^{p},mathrm{d}pi(x,x’)right)^{1/p}.
$$

For $prightarrow infty$, we have
$$W_infty(mu,nu) = lim_{p rightarrow infty} W_p(mu,nu).$$
How do I compute this in Python? I have looked at the optimal transport package, but without any luck.