update

pyg-team · Mar 12, 2024 · 28ddae0 · 28ddae0
1 parent 58c1c62
commit 28ddae0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/nn/conv/message_passing.py b/torch_geometric/nn/conv/message_passing.py
@@ -172,7 +172,7 @@ def __init__(
                     template_path=osp.join(root_dir, 'propagate.jinja'),
                     tmp_dirname='message_passing',
                     # Keyword arguments:
-                    module=self.inspector._modules,
+                    modules=self.inspector._modules,
                     collect_name='collect',
                     signature=self._get_propagate_signature(),
                     collect_param_dict=self.inspector.get_flat_param_dict(