fix log sinkhorn_ti

PythonOT · Sep 13, 2024 · 2acbdbb · 2acbdbb
1 parent 74a4ba4
commit 2acbdbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ot/unbalanced/_sinkhorn.py b/ot/unbalanced/_sinkhorn.py
@@ -1118,7 +1118,7 @@ def sinkhorn_unbalanced_translation_invariant(a, b, M, reg, reg_m, reg_type="kl"
     if n_hists:  # return only loss
         res = nx.einsum('ik,ij,jk,ij->k', u, K, v, M)
         if log:
-            return res, log
+            return res, dict_log
         else:
             return res