Unverified Commit 9ef80a6e authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[BugFix] remove non-existent arg (#6188)

parent e362f023
...@@ -276,7 +276,7 @@ class GCN2Conv(nn.Module): ...@@ -276,7 +276,7 @@ class GCN2Conv(nn.Module):
which will come into effect when printing the model. which will come into effect when printing the model.
""" """
summary = "in={_in_feats}" summary = "in={_in_feats}"
summary += ", share_weight={_share_weights}, alpha={alpha}, beta={beta}" summary += ", alpha={alpha}, beta={beta}"
if "self._bias" in self.__dict__: if "self._bias" in self.__dict__:
summary += ", bias={bias}" summary += ", bias={bias}"
if "self._activation" in self.__dict__: if "self._activation" in self.__dict__:
......
import io import io
import pickle import pickle
import random import random
import re
from copy import deepcopy from copy import deepcopy
import backend as F import backend as F
...@@ -923,6 +924,9 @@ def test_gcn2conv_e_weight(g, idtype, bias): ...@@ -923,6 +924,9 @@ def test_gcn2conv_e_weight(g, idtype, bias):
res = feat res = feat
h = gcn2conv(g, res, feat, edge_weight=eweight) h = gcn2conv(g, res, feat, edge_weight=eweight)
assert h.shape[-1] == 5 assert h.shape[-1] == 5
assert re.match(
re.compile(".*GCN2Conv.*in=.*, alpha=.*, beta=.*"), str(gcn2conv)
)
@parametrize_idtype @parametrize_idtype
......
Markdown is supported
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