From de4f7ba6fa75eac92621fba51c11bf56153603cf Mon Sep 17 00:00:00 2001 From: RĂ©mi Flamary Date: Mon, 11 Jun 2018 12:56:07 +0200 Subject: proper definition of all dor lp --- ot/lp/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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 -- cgit v1.2.3