gigl.src.inference.v1.lib.inference_output_schema#

Attributes#

`DEFAULT_EMBEDDINGS_TABLE_SCHEMA`
`DEFAULT_EMBEDDING_FIELD`
`DEFAULT_NODE_ID_FIELD`
`DEFAULT_PREDICTIONS_TABLE_SCHEMA`
`DEFAULT_PREDICTION_FIELD`
`UNENUMERATED_EMBEDDINGS_TABLE_SCHEMA`
`UNENUMERATED_PREDICTIONS_TABLE_SCHEMA`

Classes#

`InferenceOutputBigqueryTableSchema`	Thin container for inference output asset metadata
`InferenceOutputBigqueryTableSchemaBuilder`

Module Contents#

class gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema[source]#

Bases: NamedTuple

Thin container for inference output asset metadata which enables us to build and produce schemas to be fed into beam.io.WriteToBigQuery. Enables us to track the node identifier, which assists during de-enumeration.

node_field: str | None = None[source]#

schema: Dict[str, List[Dict[str, str]]] | None = None[source]#

class gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder[source]#

add_field(name, field_type, mode)[source]#

Parameters:

name (str)
field_type (str)
mode (str)

Return type:

InferenceOutputBigqueryTableSchemaBuilder

build()[source]#

Return type:: InferenceOutputBigqueryTableSchema

register_node_field(name)[source]#

Parameters:: name (str)
Return type:: InferenceOutputBigqueryTableSchemaBuilder

reset()[source]#

Return type:: None

gigl.src.inference.v1.lib.inference_output_schema.DEFAULT_EMBEDDINGS_TABLE_SCHEMA[source]#

gigl.src.inference.v1.lib.inference_output_schema.DEFAULT_EMBEDDING_FIELD = 'emb'[source]#

gigl.src.inference.v1.lib.inference_output_schema.DEFAULT_NODE_ID_FIELD = 'node_id'[source]#

gigl.src.inference.v1.lib.inference_output_schema.DEFAULT_PREDICTIONS_TABLE_SCHEMA[source]#

gigl.src.inference.v1.lib.inference_output_schema.DEFAULT_PREDICTION_FIELD = 'pred'[source]#

gigl.src.inference.v1.lib.inference_output_schema.UNENUMERATED_EMBEDDINGS_TABLE_SCHEMA[source]#

gigl.src.inference.v1.lib.inference_output_schema.UNENUMERATED_PREDICTIONS_TABLE_SCHEMA[source]#