r"""Registers a GNN prediction head in GraphGym.
(key: str, module: Any = None)
| 67 | |
| 68 | |
| 69 | def register_head(key: str, module: Any = None): |
| 70 | r"""Registers a GNN prediction head in GraphGym.""" |
| 71 | return register_base(head_dict, key, module) |
| 72 | |
| 73 | |
| 74 | def register_layer(key: str, module: Any = None): |
no test coverage detected