Merge branch 'master' into conda-forge-cmake-upstream

dmlc · Aug 22, 2023 · 6caf5f1 · 6caf5f1
2 parents 48b4ce6 + 9ef80a6
commit 6caf5f1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/python/dgl/nn/pytorch/conv/gcn2conv.py b/python/dgl/nn/pytorch/conv/gcn2conv.py
@@ -276,7 +276,7 @@ def extra_repr(self):
         which will come into effect when printing the model.
         """
         summary = "in={_in_feats}"
-        summary += ", share_weight={_share_weights}, alpha={alpha}, beta={beta}"
+        summary += ", alpha={alpha}, beta={beta}"
         if "self._bias" in self.__dict__:
             summary += ", bias={bias}"
         if "self._activation" in self.__dict__:

diff --git a/tests/python/pytorch/nn/test_nn.py b/tests/python/pytorch/nn/test_nn.py
@@ -1,6 +1,7 @@
 import io
 import pickle
 import random
+import re
 from copy import deepcopy
 
 import backend as F
@@ -923,6 +924,9 @@ def test_gcn2conv_e_weight(g, idtype, bias):
     res = feat
     h = gcn2conv(g, res, feat, edge_weight=eweight)
     assert h.shape[-1] == 5
+    assert re.match(
+        re.compile(".*GCN2Conv.*in=.*, alpha=.*, beta=.*"), str(gcn2conv)
+    )
 
 
 @parametrize_idtype