Commit 3736c5fe authored by atreyamaj's avatar atreyamaj
Browse files

Update

parent 2bf077b7
......@@ -243,8 +243,4 @@ class MATFeaturizer(MolecularFeaturizer):
adjacency_matrix = self.pad_sequence(adjacency_matrix)
distance_matrix = self.pad_sequence(distance_matrix)
node_features = np.expand_dims(node_features, 0)
adjacency_matrix = np.expand_dims(adjacency_matrix, 0)
distance_matrix = np.expand_dims(distance_matrix, 0)
return MATEncoding(node_features, adjacency_matrix, distance_matrix)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment