From 632bc9a8ec8a227d90db9635a34bb364d128cccb Mon Sep 17 00:00:00 2001
From: Hicham Janati <hicham.janati@inria.fr>
Date: Mon, 24 Jun 2019 16:19:31 +0200
Subject: update docstrings + init

---
 ot/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'ot/__init__.py')

diff --git a/ot/__init__.py b/ot/__init__.py
index acb05e6..3892d1d 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -36,4 +36,4 @@ __version__ = "0.5.1"
 __all__ = ["emd", "emd2", "sinkhorn", "sinkhorn2", "utils", 'datasets',
            'bregman', 'lp', 'tic', 'toc', 'toq', 'gromov',
            'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim',
-           'sinkhorn_unbalanced']
+           'sinkhorn_unbalanced', "barycenter_unbalanced"]
-- 
cgit v1.2.3