diff options
-rw-r--r-- | src/python/gudhi/wasserstein/__init__.py | 1 | ||||
-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: |