gigl.src.mocking.lib.mocked_dataset_resources#

Classes#

MockedDatasetInfo

Module Contents#

class gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo[source]#

get_num_edges(edge_type, edge_usage_type)[source]#

Parameters:

edge_type (gigl.src.common.types.graph_data.EdgeType)
edge_usage_type (gigl.src.common.types.graph_data.EdgeUsageType)

Return type:

int

property default_edge_type: gigl.src.common.types.graph_data.EdgeType[source]#

Return type:: gigl.src.common.types.graph_data.EdgeType

property default_node_type: gigl.src.common.types.graph_data.NodeType[source]#

Return type:: gigl.src.common.types.graph_data.NodeType

edge_dst_column_name: str = 'dst'[source]#

edge_feats: dict[gigl.src.common.types.graph_data.EdgeType, torch.Tensor] | None = None[source]#

edge_index: dict[gigl.src.common.types.graph_data.EdgeType, torch.Tensor][source]#

edge_src_column_name: str = 'src'[source]#

property edge_types: list[gigl.src.common.types.graph_data.EdgeType][source]#

Return type:: list[gigl.src.common.types.graph_data.EdgeType]

property graph_metadata_pb_wrapper: gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper[source]#

Return type:: gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper

name: str[source]#

node_feats: dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor][source]#

node_id_column_name: str = 'node_id'[source]#

node_label_column_name: str = 'node_label'[source]#

node_labels: dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor] | None = None[source]#

property node_types: list[gigl.src.common.types.graph_data.NodeType][source]#

Return type:: list[gigl.src.common.types.graph_data.NodeType]

property num_edge_features: dict[gigl.src.common.types.graph_data.EdgeType, int][source]#

Return type:: dict[gigl.src.common.types.graph_data.EdgeType, int]

property num_node_distinct_labels: dict[gigl.src.common.types.graph_data.NodeType, int][source]#

Return type:: dict[gigl.src.common.types.graph_data.NodeType, int]

property num_node_features: dict[gigl.src.common.types.graph_data.NodeType, int][source]#

Return type:: dict[gigl.src.common.types.graph_data.NodeType, int]

property num_nodes: dict[gigl.src.common.types.graph_data.NodeType, int][source]#

Return type:: dict[gigl.src.common.types.graph_data.NodeType, int]

property num_user_def_edge_features: dict[gigl.src.common.types.graph_data.EdgeType, dict[gigl.src.common.types.graph_data.EdgeUsageType, int]][source]#

Return type:: dict[gigl.src.common.types.graph_data.EdgeType, dict[gigl.src.common.types.graph_data.EdgeUsageType, int]]

sample_edge_type: gigl.src.common.types.graph_data.EdgeType | None = None[source]#

sample_node_type: gigl.src.common.types.graph_data.NodeType | None = None[source]#

task_metadata_type: gigl.src.common.types.task_metadata.TaskMetadataType[source]#

user_defined_edge_feats: dict[gigl.src.common.types.graph_data.EdgeType, dict[gigl.src.common.types.graph_data.EdgeUsageType, torch.Tensor]] | None = None[source]#

user_defined_edge_index: dict[gigl.src.common.types.graph_data.EdgeType, dict[gigl.src.common.types.graph_data.EdgeUsageType, torch.Tensor]] | None = None[source]#

version: str | None = None[source]#