Commit b17a9c17 authored by peastman's avatar peastman
Browse files

yapf

parent 82473581

deepchem/models/tensorgraph/tests/test_layers_pickle.py

+2 −2
Original line number Diff line number Diff line
@@ -418,8 +418,8 @@ def test_GraphPool_Pickle(): 
  for i in range(0, 10 + 1):

    deg_adj = Feature(shape=(None, i + 1), dtype=tf.int32)

    deg_adjs.append(deg_adj)

  layer = GraphPool(

      in_layers=[atom_features, degree_slice, membership] + deg_adjs)

  layer = GraphPool(in_layers=[atom_features, degree_slice, membership] +

                    deg_adjs)

  tg.set_loss(layer)

  tg.build()

  tg.save()

CRA Git | Maintained and supported by SUSTech CRA and CCSE