Contributor

• Tiny fix: the weights arg was not passed in the general barycenter funcs.
• Loop counter in barycenter_unbalanced was outside the loop -> change while loops to for loops
• Adding 1e-16 to avoid 0 when dual variables are returned was too conservative and causes large differences in the exponentials. changed to 1e-300

add weights arg in generic barycenter func

2f37412

same for unbalanced

a919f96

fix loop counter in barycenter + precision of dual variables

161d68a

Collaborator

Thank you Hicham !