fix test

pyg-team · Oct 20, 2022 · 357be3b · 357be3b
1 parent 1ff4fe5
commit 357be3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/nn/conv/han_conv.py b/torch_geometric/nn/conv/han_conv.py
@@ -23,7 +23,7 @@ def group(
         num_edge_types = len(xs)
         out = torch.stack(xs)
         if out.numel() == 0:
-            return out.view(0, out.size(-1))
+            return out.view(0, out.size(-1)), None
         attn_score = (q * torch.tanh(k_lin(out)).mean(1)).sum(-1)
         attn = F.softmax(attn_score, dim=0)
         out = torch.sum(attn.view(num_edge_types, 1, -1) * out, dim=0)