Commit e279904f authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #701 from rbharath/master

Removing syntax warning
parents 3682fdd5 8c3bcdd5
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -1105,7 +1105,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,