gigl.distributed.utils.serialized_graph_metadata_translator

Functions

convert_pb_to_serialized_graph_metadata(...[, ...])

Populates a SerializedGraphMetadata field from PreprocessedMetadataPbWrapper and GraphMetadataPbWrapper, containing information for loading tensors for all entities and node/edge types.

Module Contents

gigl.distributed.utils.serialized_graph_metadata_translator.convert_pb_to_serialized_graph_metadata(preprocessed_metadata_pb_wrapper, graph_metadata_pb_wrapper, tfrecord_uri_pattern='.*-of-.*\\.tfrecord(\\.gz)?$')

Populates a SerializedGraphMetadata field from PreprocessedMetadataPbWrapper and GraphMetadataPbWrapper, containing information for loading tensors for all entities and node/edge types. :param preprocessed_metadata_pb_wrapper: Preprocessed Metadata Pb Wrapper to translate into SerializedGraphMetadata :type preprocessed_metadata_pb_wrapper: PreprocessedMetadataPbWrapper :param graph_metadata_pb_wrapper: Graph Metadata Pb Wrapper to translate into Dataset Metadata :type graph_metadata_pb_wrapper: GraphMetadataPbWrapper :param tfrecord_uri_pattern: Regex pattern for loading serialized tf records :type tfrecord_uri_pattern: str

Returns:

Dataset Metadata for all entity and node/edge types.

Return type:

SerializedGraphMetadata

Parameters:

• preprocessed_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper)

• graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)

• tfrecord_uri_pattern (str)