gigl.src.data_preprocessor.lib.enumerate.utils#

Attributes#

logger

Classes#

`Enumerator`
`EnumeratorEdgeTypeMetadata`
`EnumeratorNodeTypeMetadata`

Functions#

`get_enumerated_edge_features_bq_table_name`(...)
`get_enumerated_node_features_bq_table_name`(...)
`get_enumerated_node_id_map_bq_table_name`(...)
`get_resource_labels`()

Module Contents#

class gigl.src.data_preprocessor.lib.enumerate.utils.Enumerator[source]#

run(applied_task_identifier, node_data_references, edge_data_references, gcp_project)[source]#

Parameters:

applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier)
node_data_references (Sequence[gigl.src.data_preprocessor.lib.ingest.reference.NodeDataReference])
edge_data_references (Sequence[gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference])
gcp_project (str)

Return type:

Tuple[list[EnumeratorNodeTypeMetadata], list[EnumeratorEdgeTypeMetadata]]

class gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorEdgeTypeMetadata[source]#

enumerated_edge_data_reference: gigl.src.data_preprocessor.lib.ingest.bigquery.BigqueryEdgeDataReference[source]#

input_edge_data_reference: gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference[source]#

num_edges: int[source]#

class gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorNodeTypeMetadata[source]#

bq_unique_node_ids_enumerated_table_name: str[source]#

enumerated_node_data_reference: gigl.src.data_preprocessor.lib.ingest.bigquery.BigqueryNodeDataReference[source]#

input_node_data_reference: gigl.src.data_preprocessor.lib.ingest.reference.NodeDataReference[source]#

num_nodes: int[source]#

gigl.src.data_preprocessor.lib.enumerate.utils.get_enumerated_edge_features_bq_table_name(applied_task_identifier, edge_type, edge_usage_type)[source]#

Parameters:

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

gigl.src.data_preprocessor.lib.enumerate.utils.get_enumerated_node_features_bq_table_name(applied_task_identifier, node_type)[source]#

Parameters:

applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier)
node_type (gigl.src.common.types.graph_data.NodeType)

Return type:

str

gigl.src.data_preprocessor.lib.enumerate.utils.get_enumerated_node_id_map_bq_table_name(applied_task_identifier, node_type)[source]#

Parameters:

applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier)
node_type (gigl.src.common.types.graph_data.NodeType)

Return type:

str

gigl.src.data_preprocessor.lib.enumerate.utils.get_resource_labels()[source]#

Return type:: dict[str, str]

gigl.src.data_preprocessor.lib.enumerate.utils.logger[source]#