diff options
Diffstat (limited to 'ot/__init__.py')
-rw-r--r-- | ot/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 3b072c6..5bd4bab 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -34,6 +34,7 @@ from . import stochastic from . import unbalanced from . import partial from . import backend +from . import regpath # OT functions from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d @@ -54,4 +55,4 @@ __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim', 'sinkhorn_unbalanced', 'barycenter_unbalanced', 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance', - 'smooth', 'stochastic', 'unbalanced', 'partial'] + 'smooth', 'stochastic', 'unbalanced', 'partial', 'regpath'] |