test check

Author: tvayer

ot/gromov.py

@@ -348,7 +348,7 @@ def df(G):
         log['gw_dist'] = gwloss(constC, hC1, hC2, res)
         return res, log
     else:
-        return cg(p, q, 0, 1, f, df, G0, amijo=amijo, **kwargs)
+        return cg(p, q, 0, 1, f, df, G0, amijo=amijo, C1=C1, C2=C2, constC=constC, **kwargs)
 
 
 def fused_gromov_wasserstein(M, C1, C2, p, q, loss_fun='square_loss', alpha=0.5, amijo=False, **kwargs):

ot/optim.py

@@ -73,7 +73,7 @@ def phi(alpha1):
 
 
 def do_linesearch(cost, G, deltaG, Mi, f_val,
-                  amijo=False, C1=None, C2=None, reg=None, Gc=None, constC=None, M=None):
+                  amijo=True, C1=None, C2=None, reg=None, Gc=None, constC=None, M=None):
     """
     Solve the linesearch in the FW iterations
     Parameters