gigl.src.data_preprocessor.lib.ingest.EdgeDataReference#

class gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference(reference_uri: str, edge_type: EdgeType, edge_usage_type: EdgeUsageType = EdgeUsageType.MAIN, src_identifier: str | None = None, dst_identifier: str | None = None)#

Bases: DataReference, ABC

DataReference which stores edge data

Methods

__init__

yield_instance_dict_ptransform

Returns a PTransform whose expand method returns a PCollection of InstanceDicts, which can be subsequently ingested and transformed via Tensorflow Transform.

__delattr__(name)#: Implement delattr(self, name).

__eq__(other)#: Return self==value.

__hash__()#: Return hash(self).

__init__(reference_uri: str, edge_type: EdgeType, edge_usage_type: EdgeUsageType = EdgeUsageType.MAIN, src_identifier: str | None = None, dst_identifier: str | None = None) → None#

__repr__() → str#: Return repr(self).

__setattr__(name, value)#: Implement setattr(self, name, value).

__weakref__#: list of weak references to the object (if defined)

abstract yield_instance_dict_ptransform(*args, **kwargs) → InstanceDictPTransform#

Returns a PTransform whose expand method returns a PCollection of InstanceDicts, which can be subsequently ingested and transformed via Tensorflow Transform.

TODO: extend to support multiple edge types being in the same table. :param args: :param kwargs: :return:

`__init__`
`yield_instance_dict_ptransform`	Returns a PTransform whose expand method returns a PCollection of InstanceDicts, which can be subsequently ingested and transformed via Tensorflow Transform.