summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortlacombe <lacombe1993@gmail.com>2020-03-31 15:02:32 +0200
committertlacombe <lacombe1993@gmail.com>2020-03-31 15:02:32 +0200
commit842475615841f864b4ce41a2a4b69f1e189a2946 (patch)
tree3102227e3be0bb019b3baabaf79b575b260ff21c
parent1aaffd2e1fab45988d92f5e51a9d294696ff5492 (diff)
created wasserstein repo
-rw-r--r--src/python/gudhi/wasserstein/__init__.py1
-rw-r--r--src/python/gudhi/wasserstein/barycenter.py (renamed from src/python/gudhi/barycenter.py)0
-rw-r--r--src/python/gudhi/wasserstein/wasserstein.py (renamed from src/python/gudhi/wasserstein.py)2
3 files changed, 2 insertions, 1 deletions
diff --git a/src/python/gudhi/wasserstein/__init__.py b/src/python/gudhi/wasserstein/__init__.py
new file mode 100644
index 00000000..ed225ba4
--- /dev/null
+++ b/src/python/gudhi/wasserstein/__init__.py
@@ -0,0 +1 @@
+from .wasserstein import wasserstein_distance
diff --git a/src/python/gudhi/barycenter.py b/src/python/gudhi/wasserstein/barycenter.py
index 079bcc57..079bcc57 100644
--- a/src/python/gudhi/barycenter.py
+++ b/src/python/gudhi/wasserstein/barycenter.py
diff --git a/src/python/gudhi/wasserstein.py b/src/python/gudhi/wasserstein/wasserstein.py
index 51d1d83c..e1233eec 100644
--- a/src/python/gudhi/wasserstein.py
+++ b/src/python/gudhi/wasserstein/wasserstein.py
@@ -9,7 +9,7 @@
import numpy as np
import scipy.spatial.distance as sc
-import gudhi.barycenter as barycenter
+
try:
import ot
except ImportError: