diff options
Diffstat (limited to 'ot/__init__.py')
-rw-r--r-- | ot/__init__.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index dd74590..f63d1c2 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -5,12 +5,14 @@ import utils import datasets import plot import bregman +import da # OT functions from emd import emd from bregman import sinkhorn,barycenter +from da import sinkhorn_lpl1_mm # utils functions from utils import dist,unif -__all__ = ["emd","sinkhorn","utils",'datasets','bregman','plot','dist','unif','barycenter'] +__all__ = ["emd","sinkhorn","utils",'datasets','bregman','plot','dist','unif','barycenter','sinkhorn_lpl1_mm'] |