diff options
Diffstat (limited to 'ot/__init__.py')
-rw-r--r-- | ot/__init__.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 0602eed..72c820a 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -1,22 +1,21 @@ -# Python Optimal Transport toolbox +"""Python Optimal Transport toolbox""" # All submodules and packages -from . import lp +from . import lp from . import bregman -from . import optim +from . import optim from . import utils from . import datasets from . import plot from . import da - - # OT functions from .lp import emd -from .bregman import sinkhorn,barycenter +from .bregman import sinkhorn, barycenter from .da import sinkhorn_lpl1_mm # utils functions -from .utils import dist,unif +from .utils import dist, unif -__all__ = ["emd","sinkhorn","utils",'datasets','bregman','lp','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim'] +__all__ = ["emd", "sinkhorn", "utils", 'datasets', 'bregman', 'lp', 'plot', + 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim'] |