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. Args: preprocessed_metadata_pb_wrapper (PreprocessedMetadataPbWrapper): Preprocessed Metadata Pb Wrapper to translate into SerializedGraphMetadata graph_metadata_pb_wrapper (GraphMetadataPbWrapper): Graph Metadata Pb Wrapper to translate into Dataset Metadata tfrecord_uri_pattern (str): Regex pattern for loading serialized tf records Returns: SerializedGraphMetadata: Dataset Metadata for all entity and node/edge types.