update

pyg-team · May 27, 2023 · 7e809ab · 7e809ab
1 parent dccc482
commit 7e809ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/hetero/hierarchical_sage.py b/examples/hetero/hierarchical_sage.py
@@ -40,8 +40,8 @@ def __init__(self, edge_types, hidden_channels, out_channels, num_layers):
 
         self.lin = Linear(hidden_channels, out_channels)
 
-    def forward(self, x_dict, edge_index_dict, num_sampled_edges_dict=None,
-                num_sampled_nodes_dict=None):
+    def forward(self, x_dict, edge_index_dict, num_sampled_edges_dict,
+                num_sampled_nodes_dict):
 
         for i, conv in enumerate(self.convs):
             x_dict, edge_index_dict, _ = trim_to_layer(