summaryrefslogtreecommitdiff
path: root/ot/lp/__init__.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-06-11 12:56:07 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-06-11 12:56:07 +0200
commitde4f7ba6fa75eac92621fba51c11bf56153603cf (patch)
treec93a843a7eaf597953d68f6b72ebb117214e6426 /ot/lp/__init__.py
parentf22be34f5340525d195b643cde29344f719a7c33 (diff)
proper definition of all dor lp
Diffstat (limited to 'ot/lp/__init__.py')
-rw-r--r--ot/lp/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py
index 5dda82a..4c0d170 100644
--- a/ot/lp/__init__.py
+++ b/ot/lp/__init__.py
@@ -18,6 +18,8 @@ from .emd_wrap import emd_c, check_result
from ..utils import parmap
from .cvx import barycenter
+__all__=['emd', 'emd2', 'barycenter', 'cvx']
+
def emd(a, b, M, numItermax=100000, log=False):
"""Solves the Earth Movers distance problem and returns the OT matrix