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