Update new.py (#404)

aiqm · Dec 13, 2019 · abc8f7f · abc8f7f
1 parent be84845
commit abc8f7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchani/data/new.py b/torchani/data/new.py
@@ -593,7 +593,7 @@ def collate_fn(data, chunk_threshold, properties_info):
         if properties_info['padding_values'][i] is None:
             prop = torch.stack(prop)
         else:
-            prop = torch.nn.utils.rnn.pad_sequence(batch_species,
+            prop = torch.nn.utils.rnn.pad_sequence(prop,
                                                    batch_first=True,
                                                    padding_value=properties_info['padding_values'][i])
         # sort with number of atoms