diff options
| author | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 16:40:06 +0200 |
|---|---|---|
| committer | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 16:40:06 +0200 |
| commit | 897982718a5fd81a9a591d80a7d50839399fc088 (patch) | |
| tree | 48189493c09cda25ee19dfd0b7ef59c2f6819ba7 /ot/__init__.py | |
| parent | 50bc90058940645a13e2f3e41129bdc97161dc63 (diff) | |
fix func names + add more tests
Diffstat (limited to 'ot/__init__.py')
| -rw-r--r-- | ot/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 361be02..acb05e6 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -25,7 +25,7 @@ from . import unbalanced # OT functions from .lp import emd, emd2 from .bregman import sinkhorn, sinkhorn2, barycenter -from .unbalanced import sinkhorn_unbalanced +from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced from .da import sinkhorn_lpl1_mm # utils functions |