Add partial_wasserstein function to API

Introduces the partial_wasserstein function to the Julia API, wrapping pot.partial.partial_wasserstein2. Also updates the package version to 0.1.9 and exports the new function.

Author: pnavaro

Project.toml

@@ -1,7 +1,7 @@
 name = "PythonOT"
 uuid = "3c485715-4278-42b2-9b5f-8f00e43c12ef"
 authors = ["David Widmann"]
-version = "0.1.8"
+version = "0.1.9"
 
 [deps]
 PyCall = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0"

src/PythonOT.jl

@@ -15,6 +15,7 @@ export emd,
     empirical_sinkhorn_divergence,
     mm_unbalanced,
     entropic_partial_wasserstein,
+    partial_wasserstein,
     partial_wasserstein2
 
 const pot = PyCall.PyNULL()

src/lib.jl

@@ -587,6 +587,10 @@ function entropic_partial_wasserstein(a, b, M, reg; kwargs...)
     return pot.partial.entropic_partial_wasserstein(a, b, M, reg; kwargs...)
 end
 
+function partial_wasserstein(a, b, M; kwargs...)
+    return pot.partial.partial_wasserstein2(a, b, M; kwargs...)
+end
+
 function partial_wasserstein2(a, b, M; kwargs...)
     return pot.partial.partial_wasserstein2(a, b, M; kwargs...)
 end