Update __init__.py

drorlab · Jul 14, 2021 · 82af6b2 · 82af6b2
1 parent dff69f2
commit 82af6b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gvp/__init__.py b/gvp/__init__.py
@@ -329,7 +329,7 @@ def forward(self, x, edge_index, edge_attr,
         :param edge_index: array of shape [2, n_edges]
         :param edge_attr: tuple (s, V) of `torch.Tensor`
         :param autoregressive_x: tuple (s, V) of `torch.Tensor`. 
-                If not `None`, will be used as srcqq node embeddings
+                If not `None`, will be used as src node embeddings
                 for forming messages where src >= dst. The corrent node 
                 embeddings `x` will still be the base of the update and the 
                 pointwise feedforward.
@@ -371,4 +371,4 @@ def forward(self, x, edge_index, edge_attr,
         if node_mask is not None:
             x_[0][node_mask], x_[1][node_mask] = x[0], x[1]
             x = x_
-        return x
+        return x