From e55232056a79de128583b87e65abc6d7a75fb298 Mon Sep 17 00:00:00 2001
From: Hicham Janati <hicham.janati@inria.fr>
Date: Tue, 3 Sep 2019 18:15:25 +0200
Subject: add unbalanced_barycenter import

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

(limited to 'ot')

diff --git a/ot/__init__.py b/ot/__init__.py
index 7d9615a..99f288e 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -65,7 +65,7 @@ from . import unbalanced
 # OT functions
 from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d
 from .bregman import sinkhorn, sinkhorn2, barycenter
-from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced
+from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced, sinkhorn_unbalanced2
 from .da import sinkhorn_lpl1_mm
 
 # utils functions
-- 
cgit v1.2.3