Commit 8c3bcdd5 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Making error message vanish

parent e5475d1c

deepchem/models/tensorgraph/layers.py

+1 −1
Original line number Diff line number Diff line
@@ -1091,7 +1091,7 @@ class GraphGather(Layer): 


    # Perform the mol gather



    assert (self.batch_size > 1, "graph_gather requires batches larger than 1")

    assert self.batch_size > 1, "graph_gather requires batches larger than 1"



    # Obtain the partitions for each of the molecules

    activated_par = tf.dynamic_partition(atom_features, membership,

CRA Git | Maintained and supported by SUSTech CRA and CCSE