Merge pull request #188 from hl2500/master

Update featurization.py
chemprop · Aug 10, 2021 · 4b00fbf · 4b00fbf
2 parents 3f20ca4 + 56f8209
commit 4b00fbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chemprop/features/featurization.py b/chemprop/features/featurization.py
@@ -510,7 +510,7 @@ def get_a2a(self) -> torch.LongTensor:
         """
         Computes (if necessary) and returns a mapping from each atom index to all neighboring atom indices.
 
-        :return: A PyTorch tensor containing the mapping from each bond index to all the incoming bond indices.
+        :return: A PyTorch tensor containing the mapping from each atom index to all the neighboring atom indices.
         """
         if self.a2a is None:
             # b = a1 --> a2