def __init__(
self,
module: Module,
metadata: Metadata,
num_bases: int,
in_channels: Optional[Dict[str, int]] = None,
input_map: Optional[Dict[str, str]] = None,
debug: bool = False,
):
super().__init__(module, input_map, debug)
unused_node_types = get_unused_node_types(*metadata)
if len(unused_node_types) > 0:
warnings.warn(
f"There exist node types ({unused_node_types}) whose "
f"representations do not get updated during message passing "
f"as they do not occur as destination type in any edge type. "
f"This may lead to unexpected behaviour.")
self.metadata = metadata
self.num_bases = num_bases
self.in_channels = in_channels or {}
assert len(metadata) == 2
assert len(metadata[0]) > 0 and len(metadata[1]) > 0
# Compute IDs for each node and edge type:
self.node_type2id = {k: i for i, k in enumerate(metadata[0])}
self.edge_type2id = {k: i for i, k in enumerate(metadata[1])}
def __init__(
self,
module: Module,
metadata: Metadata,
aggr: str = 'sum',
input_map: Optional[Dict[str, str]] = None,
debug: bool = False,
):
super().__init__(module, input_map, debug)
unused_node_types = get_unused_node_types(*metadata)
if len(unused_node_types) > 0:
warnings.warn(
f"There exist node types ({unused_node_types}) whose "
f"representations do not get updated during message passing "
f"as they do not occur as destination type in any edge type. "
f"This may lead to unexpected behaviour.")
self.metadata = metadata
self.aggr = aggr
assert len(metadata) == 2
assert len(metadata[0]) > 0 and len(metadata[1]) > 0
assert aggr in self.aggrs.keys()