Commit ffa3fe1f authored by Atreya Majumdar's avatar Atreya Majumdar
Browse files

Changed result zeros parameter

parent 8535d72d
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -97,7 +97,7 @@ class MATFeaturizer(MolecularFeaturizer):
    result = node_features

    result = np.zeros((node_features.shape[0],
                       node_features.shape[1] + 2 * adjacency_matrix.shape[1]))
                       node_features.shape[1] + adjacency_matrix.shape[1] + distance_matrix.shape[1]))

    for i in range(node_features.shape[0]):
      result[i, :node_features.shape[1]] = node_features[i]