fix pep8 and tests

Author: cedricvincentcuaz

ot/gromov/_gw.py

@@ -167,7 +167,7 @@ def df(G):
             return 0.5 * (gwggrad(constC, hC1, hC2, G, np_) + gwggrad(constCt, hC1t, hC2t, G, np_))
 
     # removed since 0.9.2
-    #if loss_fun == 'kl_loss':
+    # if loss_fun == 'kl_loss':
     #    armijo = True # there is no closed form line-search with KL
 
     if armijo:
@@ -479,7 +479,7 @@ def df(G):
             return 0.5 * (gwggrad(constC, hC1, hC2, G, np_) + gwggrad(constCt, hC1t, hC2t, G, np_))
 
     # removed since 0.9.2
-    #if loss_fun == 'kl_loss':
+    # if loss_fun == 'kl_loss':
     #    armijo = True  # there is no closed form line-search with KL
 
     if armijo:
@@ -828,7 +828,7 @@ def gromov_barycenters(
         C = init_C
 
     # removed since 0.9.2
-    #if loss_fun == 'kl_loss':
+    # if loss_fun == 'kl_loss':
     #    armijo = True
 
     cpt = 0
@@ -1015,7 +1015,7 @@ def fgw_barycenters(
         T = [nx.outer(p, q) for q in ps]
 
     # removed since 0.9.2
-    #if loss_fun == 'kl_loss':
+    # if loss_fun == 'kl_loss':
     #    armijo = True
 
     cpt = 0

test/test_gromov.py

@@ -1431,7 +1431,7 @@ def test_fgw_barycenter(nx):
             alpha=0.5, fixed_structure=True, init_C=None, fixed_features=False,
             p=None, loss_fun='square_loss', max_iter=100, tol=1e-3
         )
-    except:
+    except ot.utils.UndefinedParameter:
         Xb, Cb = ot.gromov.fgw_barycenters(
             n_samples, [ysb, ytb], [C1b, C2b], ps=[p1b, p2b], lambdas=None,
             alpha=0.5, fixed_structure=True, init_C=init_Cb, fixed_features=False,
@@ -1451,7 +1451,7 @@ def test_fgw_barycenter(nx):
             p=pb, loss_fun='square_loss', max_iter=100, tol=1e-3,
             warmstartT=True, log=True, random_state=98765, verbose=True
         )
-    except:
+    except ot.utils.UndefinedParameter:
         Xb, Cb, logb = ot.gromov.fgw_barycenters(
             n_samples, [ysb, ytb], [C1b, C2b], [p1b, p2b], [.5, .5], 0.5,
             fixed_structure=False, fixed_features=True, init_X=init_Xb,