Add POT optimal transport option in the default numpy version

[dimitra-maoutsa]

Add if clause that checks if the environment is POTTED, and reweight with the POT OT call.

The torched version has only the POTTED option, but that's fine...