gigl.src.common.translators.gbml_protos_translator#

Attributes#

Classes#

GbmlProtosTranslator

Module Contents#

class gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator[source]#

static EdgeTypePb_from_edge_type(edge_type)[source]#

Parameters:: edge_type (gigl.src.common.types.graph_data.EdgeType)
Return type:: snapchat.research.gbml.graph_schema_pb2.EdgeType

static edge_from_EdgePb(graph_metadata_pb_wrapper, edge_pb)[source]#

Parameters:

graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)
edge_pb (snapchat.research.gbml.graph_schema_pb2.Edge)

Return type:

Tuple[gigl.src.common.types.graph_data.Edge, Optional[torch.Tensor]]

static edge_type_from_EdgeTypePb(edge_type_pb)[source]#

Parameters:: edge_type_pb (snapchat.research.gbml.graph_schema_pb2.EdgeType)
Return type:: gigl.src.common.types.graph_data.EdgeType

static graph_data_from_GraphPb(samples, graph_metadata_pb_wrapper, builder)[source]#

Parameters:

samples (list[snapchat.research.gbml.graph_schema_pb2.Graph])
graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)
builder (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder)

Return type:

gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol

static node_from_NodePb(graph_metadata_pb_wrapper, node_pb)[source]#

Parameters:

graph_metadata (GraphMetadataPbWrapper)
node_pb (graph_schema_pb2.Node)
graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)

Returns:

Tuple of Node and related Node features

Return type:

Tuple[Node, torch.tensor]

gigl.src.common.translators.gbml_protos_translator.logger[source]#