fix tests

Author: cedricvincentcuaz

ot/gromov/_semirelaxed.py

@@ -1081,4 +1081,4 @@ def entropic_semirelaxed_fused_gromov_wasserstein2(
     if log:
         return log_srfgw['srfgw_dist'], log_srfgw
     else:
-        return log_srfgw['srfgw_dist']
+        return log_srfgw['srfgw_dist']

ot/gromov/_utils.py

@@ -361,7 +361,7 @@ def update_kl_loss(p, lambdas, T, Cs, nx=None):
     # Correct order mistake in Equation 15 in [12]
     tmpsum = sum([
         lambdas[s] * nx.dot(
-            nx.dot(T[s], nx.log(Cs[s])),
+            nx.dot(T[s], nx.log(nx.maximum(Cs[s], 1e-15))),
             T[s].T
         ) for s in range(len(T))
     ])