Add partial_wasserstein2 function and export

Introduces the partial_wasserstein2 function in lib.jl, wrapping the corresponding PythonOT method, and exports it in PythonOT.jl for external use.

Author: pnavaro

src/PythonOT.jl

@@ -14,7 +14,8 @@ export emd,
     sinkhorn_unbalanced2,
     empirical_sinkhorn_divergence,
     mm_unbalanced,
-    entropic_partial_wasserstein
+    entropic_partial_wasserstein,
+    partial_wasserstein2
 
 const pot = PyCall.PyNULL()
 

src/lib.jl

@@ -586,3 +586,7 @@ julia> round.(entropic_partial_wasserstein(a, b, M, 1, m=0.1), digits=2)
 function entropic_partial_wasserstein(a, b, M, reg; kwargs...)
     return pot.partial.entropic_partial_wasserstein(a, b, M, reg; kwargs...)
 end
+
+function partial_wasserstein2(a, b, M; kwargs...)
+    return pot.partial.partial_wasserstein2(a, b, M; kwargs...)
+end